Adding naive bayes