Update README.md
Adding some basic unit tests