Get treemap to work as a widget.