Fixing errors in run.py