add feature_importances
remove comments