Start adding unit tests + Mockito dependencies