Add function classification for testing Naive Bayes