added more cases to naive bayes model
added a bunch of stuff