Update main.cpp

thank you.
you can test it