some tweaks to persistence_tools/diagrams