updates to add code coverage