Add cancellation token feature for docs