Update naive_bayes.js