Update assignment.scala