Added tests to increase code coverage