better class diagram 2