WIP creating naive bayes model for classification