Update main.cpp
Merge pull request #3 from shramdas/patch-1

Update main.cpp