Updated test 1/2 to use controller, updated uml png