Update prepare_data1.py