Cleaned run.py a little bit