Tweaking the argparse snippet