cleans up main.scala