updated class diagrams