Add command line parser (with argparse)