Update DAGScheduler.scala