Updated README and unit tests