Update Lab1.scala

Fixed some things causing test errors