Adding logic for highlighting header