Update Working with more diagrams