Create prepare_data.py