Add code for highlighting text