Updating main.scala