Update README.md
add unit tests