Changed class diagram