Update gen_data.sh