added variable importance, minor corrections