Fixing Naive Bayes R Script