added more test code coverage