Quick refactor of run.py using methods for argparse and main