attempt at our UML for project 2