Update NaiveBayes.scala