add argparse and updated readme.