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

Update main.c