Update/text highlight

Updated this feature to add text highlighting to individual elements, not the parent container.