changes in highlight function

faster highlighting