Update main.c
Merge pull request #3 from hhxuliang/master

Update main.c