Added naive bayes classifier