Update prepare_data4_2-split-A.py