Adding support for callback in docs