just added the argparse functionality