added naive bayes calculations, needs testing now