Update Gradient.scala