modify hightlight code