Semi-working Naive Bayes, needs to be tested