Clean up run script, shorter code, more abstraction.  Use argparse.