Added code coverage to tests