Update prepare_data2.py