Make balance = observations
work with strata
#364
Merged
balance = observations
work with strata
#364