Used argparse to parse the arguments