Update README.md and bug fix