Add test coverage support