Made run.py more sane, i think