Update prepare_data3_2-split.py