minor changes to naive bayes