Use argparse for command line arguments