added naive bayes