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

Update main.c