Update main.c
Merge pull request #5 from ssriva12/patch-2

Update main.c