Added UML for controller