bug fix & update README.md