Add header to main.c

Signed-off-by: oooxff <oooxff@gmail.com>