Update run.py