Add code hightlighting