Use argparse module for argument parsing