Modified. Added functionality with argparse.