Improving code coverage