made some updates to run.py