Update of class diagram