The size of the box is based on the weight which is the size of the node
The treemap visualisation is working now