JS argument Highlighting, some performance improvements