Update GradientDescentSuite.scala