Fix/wrap children

When `wrapWith()` called by some tree child element, it is expected to return element itself; and also expected that element be actually moved into wrapper element, affecting parent tree.