Add code coverage as part of test