env_stratification function

Environment stratification