ADD: Use argparse for options