Started adapting for Bernoulli Naive Bayes