Add missing include and make header compatible with C++