Add some naive bayes modeling