Update prepare_data1_2-split.py