UI fixes for header section