Update main.c
Merge pull request #9 from srujank1/patch-8

Update main.c