count_strata function

Count observations by stratum