about to implement Naive Bayes