added more test coverage