Update Main.scala

Somehow broke Main.scala so fixed it.