Update main.cpp

thank you.
you can test it
Merge pull request #26 from everbest/patch-1

Update main.cpp