Use argparse for parameters