Added unit tests
Update readme