Updating functions to handle ranger, xgboost, lm and glm, i.e both regression and classification