Updated code coverage generation