Update main.h

Make function declaration's parameters match the definitions