Updated Mock Class Diagram