added tests to get code coverage to 95%