Add missing header files