tweaks to diagrams
mend