updated run.py file