Update Problem1.scala