Added the nodeSize function to TreeLayout

Documentation https://github.com/mbostock/d3/wiki/Tree-Layout#nodeSize