Add missing #include's