Added TODO to utilize argparse module.