Updated text highlighting part