Added the naive bayes predictor