Added code to highlight words