fixing naive bayes for two variables