Added unit tests and updated Readme