Added naive bayes and plots. improved structure