revised code for code hightlighting