Adding some tests and using mocks