Fourth part of task:
-introducing mocks