Update main.cpp
Merge pull request #5 from CristanIonescu/patch-5

Update main.cpp