Add support for code coverage