diff --git a/src/App.jsx b/src/App.jsx
index e68c527..334ad27 100644
--- a/src/App.jsx
+++ b/src/App.jsx
@@ -211,6 +211,14 @@ function App() {
     }, [flowInstance, setNodes]
   )
 
+  const onNodesDelete = useCallback((deletedNodes) => {
+    setNodes((nds) => nds.filter((node) => !deletedNodes.some((dn) => dn.id === node.id)))
+  }, [setNodes])
+
+  const onEdgesDelete = useCallback((deletedEdges) => {
+    setEdges((eds) => eds.filter((edge) => !deletedEdges.some((de) => de.id === edge.id)))
+  }, [setEdges])
+
   function handleSaveModalData(formData) {
     if (!modalData)
       return
@@ -947,6 +955,9 @@ function App() {
               onDrop = {onDrop}
               onDragOver = {onDragOver}
               isValidConnection = {isValidConnection}
+              onNodesDelete = {onNodesDelete}
+              onEdgesDelete = {onEdgesDelete}
+              deleteKeyCode = {["Delete", "Backspace"]}
               //fitView
             >
               <Controls/>