Adding class diagrams