Add missing headers in some files