generate_subgroup function

Generate Dataset with different treatment effect in subgroup