Got highlighting current sub element working