Update prepare_data4_2b.py