change the class diagram