update lab1.scala