Add cancellation token to simulation