修改了bug,将main一个代码写成函数