public void handleDisconnect(Figure start, Figure end) { PertFigure source = (PertFigure) start; PertFigure target = (PertFigure) end; if (target != null) { target.removePreTask(source); target.updateDurations(); } if (source != null) { source.removePostTask(target); } }