add gen_data.py