Added mockito and created a test