forgot updated class diagram.