Add types for job cancellation token