update gen_data.py