Update readme
Add unit tests