Added code coverage to help see what needs to be tested