Added UML for Assignment 5, fixed visibility