Update YarnClusterScheduler.scala