Switch to using argparse