Add CLI argument parsing with argparse.