Naive Bayes clean up, added random forest classification