Update BagOfWords.scala