adding test coverage for all the classes