doc: Extend the class diagram