fix highlight button function