Added the nodeSize function to TreeLayout

See also the [d3 documentation](https://github.com/mbostock/d3/wiki/Tree-Layout#nodeSize)