Adding back UML and UML classes diagram