run.py updated