added code coverage for tests