Update run.py

still rewriting functions in Python... :)