Update main.scala