Update the task models to use the IScheduler interface