Gini importance correct, need refactoring now