class diagram uppdated