Added naive bayes trainer