Move argparse code under conditional
PEP8