updated naive_bayes