Add button highlight function