Adding Bayes Naive support