improve the code highlight

Signed-off-by: Xiaofeng QU <xiaofeng.qu.hk@ieee.org>