Minor update to the headers