added tests for more coverage