Update main.cpp

调整一下代码而已
Merge pull request #1 from lumm/patch-1

Update main.cpp