class diagrams fix