working on increasing test coverage