Added two unit tests using the Mockito framework