Update prepare.py