Convert argument parsing to argparse