Modified h2o.randomForest