Update highlight_words command to take arguments.