added unit test coverage