GF: Improve multi-category predictive editing