Update HW2.scala