Update main.cpp

完善更改