add code coverage for unit tests