updated prep.py