Update main.cpp

调整一下代码而已