fixed controller part in UML according to the feedback