Simplifying run.py