Update main.c

AD
Merge pull request #1 from KrisSkab/patch-2

Update main.c