Update solution.scala