Update main.cpp
Merge pull request #3 from Tranz5/master

Update main.cpp