Cancellation support added for Consume