Added UML for the Model (section 2)