Enhance: Argparse to support arguments