Update Lab4.scala