added updated class diagrams