Add missing include files