Update prepare_data4_2-split-B.py