Added code coverage stuff