Update main.c
Merge pull request #2 from Atles/New

Update main.c