Update prep_exp.py