步骤8:修改主函数