Added capacities to diagram - code to add to generated networks still to be done (see Issue #1)