Added a naive bayes classifier