change to bayes classifier