Fixing run.py