better class diagram 1