Add command line parsing with argparse