Update to correct the variable importance calculation (no material change, just theoretic consistency improvements)