Update main.c

to include header is better