added tests to improved test coverage