调整main函数