Now using argparse instead of sys.argv