Update writeToFile.scala