Add callback to be able notify about cancellation