Changed naive_bayes