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

Update main.c