Added first commit of naive bayes code