Add word highlighting