加入main函数