修改main函数bug