completed naive bayes system but predictions still not as expected