Update run.py

small changes