Added UML for class diagrams