modify the class diagram