Minor changes (different arg name for argparse)