gen_data function

Generate TWFE data