private void saveNetworkElementChangement() {
    simulationBox.endEdition();
    if (simulationBox.isElementModified()) {
      if (simulationBox.isAnimationModified()) {

      } else {
        SimulationElement modifiedElement = simulationBox.getModifiedElement();
        if (isTransit(modifiedElement)) {
          Transit modifiedTransit = (Transit) modifiedElement;
          currentTransit.modifyTransit(modifiedTransit.getIdentifier(), modifiedTransit);
        } else {
          currentNetwork.modifyNetworkElement(modifiedElement.getIdentifier(), modifiedElement);
          if (simulationBox.isThereASecondSegment()) {
            Segment secondSegment = simulationBox.getModifiedSecondWay();
            currentNetwork.modifyNetworkElement(secondSegment.getIdentifier(), secondSegment);
          }
        }
        updateNetworkData();
        simulationBox.refreshElementsAfterModification();
      }
    }
  }
 private boolean isTransit(SimulationElement modifiedElement) {
   return modifiedElement.getClass().getSimpleName().equals(TransportNeed.class.getSimpleName())
       || modifiedElement.getClass().getSimpleName().equals(Circuit.class.getSimpleName());
 }