Adding missing #includes