Adding missing include directives