Add mocks and stubs