working, but how should the diagram look?