Switch from getopt to argparse