增加分类逻辑

增加分类逻辑