[feature] 修复分类bug。升级分类功能