Using argparse to get command-line input