Navie bayes working version