改进highlight_code方法