Improvements to dragging from WidgetTree