Convert challenge2 to use argparse