Added argparse functionality