Add support for coverage testing