update prep.py