Update run.py in an attempt to fix issues