modified run.py