Added class diagrams