完成主函数

完成主函数，实现分类功能