Add support for callback chaining