add strata count function