Add highlight state for buttons