Update prepare_data.py