add code coverage tests