Use `argparse` module for argument parsing