updated run.py script