class diagram relations fix