Moved the selected state change color inside the header button