Add missing includes