adding in code coverage stuff