Modify DIAGRAM for prettier printing

Diagram printing tool isn't that smart,
so based on what the input diagram looks like,
the output will change (semantically the same
output, just displayed differently)