Clean up run.py