Changes in class diagram