Update prepare_data4_2.py