Add test coverage, update readme