add unit test coverage