Remove unnecessary comments
Add some more diagrams