Add auto highlight variables function