@pravnar can you please take a look at the code above to make sure I am not doing anything dumb and also at the naivebayes runner and verify