Add code coverage annotations to tests