Added code coverage to test