fixed a few stuff in run.py