Cleaned up

cleaned up lab2.scala