Implemented highlighting elements in tree (Test)