js HighLight code enhancement