add wires back in and style the box diagrams