Update WordCount.scala