Updated README, added unit tests