Naive Bayes Code as a function_wootwoot