Add code coverage testing