added argparse and updated readme.