Update main.cpp
Merge pull request #3 from Hoangngo-dev/patch-5

Update main.cpp