Fix a minor bug and update readme.md