added missed function
Added naive Bayes classifier