some tweaks to the model and playing at naiive bayes classification