Create prepare_output.py