ENH: The network output now considers expansion that occurs in individual layers.
ENH: Bug fixes in the visualization code.