Updated Class Diagram

updated class diagram