Update ApplicationMaster.scala