Update TaskScheduler.scala