run.py modifications