[SANDBOX] improved feature_grouping.py