strat_data function

Prepare simulated data for stratified analysis