update code of run.py