Update prepare_data2-split.py