Add UML for Lab 4