Update Lab2.scala