Update exercise.scala