adding mocking of other functions