clean up run.py