Update prepare_data3_2.py