run.py is a little more DRY