done building naive bayes, need to test it