enhance the js code highlight