Adding a missing method and importing the mockito dependancy