Add missing includes directives