added mockito, started with writing tests