add initial version of code for Naive Bayes class