Added callback support for in mocking