Update README.md and unit test