Clarifying code and adding todos for feature importances