updated class diagram .png