Revert "Revert "Class diagram quick update""

This reverts commit c07a610101c4ad977412902645bdefeedda11247.