Update prepare_data4.py