Add code coverage - closes #3