Add cancellation_token