- add partitioning magic for fair distribution of features with many words