[Update] add highlight for code