Update main.cpp

fix compiler warning
Merge pull request #20 from fulldecent/patch-1

Update main.cpp