Add missing include in main.c