adding more unit tests, updating readme