MultinomialNB base test; Added some functionality to RDD wrappers