Add missing headers