Update main.cpp
Merge pull request #4 from Hoangngo-dev/patch-6

Update main.cpp