Include missing headers