added more class diagrams