Update prepare_data1-split.py