Added some more unit tests and updated the readme