Updated CancellationToken parameter