Update naive_bayes.rb