Next iteration to further group by stratification id in addition to stratified/not-stratified