Clean up the diagram layer.
Removing the second integer output since that just has weird side effects.