Added mockito classes for