adding more test coverage