Added two class diagrams