add in naive bayes