NaiveBayes implementation works for first example at least