Add missing #include files