Add code coverage for test suite