Cleanup `diagrams` tool code to account for protocol properties such as `calls/2` that are no longer supported