Rewrite Class Diagram