class diagram modified