Major class diagram update