Added highlight function