Fix bug on 3D diagrams
Add more diagrams to 3D example