updated run.py, still needs work