Clean UML schema, update documentation