Updated code for header customization