Add highlight for function/class etc.