Small update to header