Cleaning up argparse setup