class diagram changes