Updated README and cleaned up unit test