Updated with BernoulliNB