Added an implementation of class mocks