SImple training using naive bayes