show inheritance arrows when dragging classes in diagram

I found that the functionality to drag element in the diagram works almost perfect, especially for objects.

In the case of classes, the main thing missing are the inheritance arrows.

If that could be fixed, would be great!