First stab at Naive bayes