initial naive bayes work