added tests for complete code coverage