Update main.h

Make function declaration's parameters match the definitions
Merge pull request #5 from ColorfulGecko/patch-5

Update main.h