first draft of naive bayes