diff --git a/frontend/src/lib/components/flows/map/FlowModuleSchemaMap.svelte b/frontend/src/lib/components/flows/map/FlowModuleSchemaMap.svelte
index f3b593177e..0608985720 100644
--- a/frontend/src/lib/components/flows/map/FlowModuleSchemaMap.svelte
+++ b/frontend/src/lib/components/flows/map/FlowModuleSchemaMap.svelte
@@ -289,8 +289,6 @@
 				}
 			}}
 			on:insert={async ({ detail }) => {
-				console.log('insert', detail)
-
 				if (shouldRunTutorial('forloop', detail.detail, 1)) {
 					flowTutorials?.runTutorialById('forloop', detail.index)
 				} else if (shouldRunTutorial('branchone', detail.detail, 2)) {
diff --git a/frontend/src/lib/components/graph/FlowGraphV2.svelte b/frontend/src/lib/components/graph/FlowGraphV2.svelte
index 3917ba858d..9df74199a6 100644
--- a/frontend/src/lib/components/graph/FlowGraphV2.svelte
+++ b/frontend/src/lib/components/graph/FlowGraphV2.svelte
@@ -203,7 +203,6 @@
 			if ($selectedId != mod.id) {
 				$selectedId = mod.id
 			}
-			console.log('selected', mod)
 			dispatch('select', mod)
 		}
 	}
diff --git a/frontend/src/lib/components/graph/checklist.md b/frontend/src/lib/components/graph/checklist.md
deleted file mode 100644
index 8f669b5463..0000000000
--- a/frontend/src/lib/components/graph/checklist.md
+++ /dev/null
@@ -1,11 +0,0 @@
-## Rendering
-
-- Nodes [x]
-- Edges [x]
-- Insert button [x]
-- AI button [x]
-- Delete branch button [x]
-- Insert branch button [x]
-- Graph actions [x]
-
-## Functionality
diff --git a/frontend/src/lib/components/graph/renderers/edges/BaseEdge.svelte b/frontend/src/lib/components/graph/renderers/edges/BaseEdge.svelte
index 8141cb860f..907bd9eaf5 100644
--- a/frontend/src/lib/components/graph/renderers/edges/BaseEdge.svelte
+++ b/frontend/src/lib/components/graph/renderers/edges/BaseEdge.svelte
@@ -52,7 +52,14 @@
 			class="edgeButtonContainer nodrag nopan"
 			style:transform="translate(-50%, -50%) translate({labelX}px,{labelY}px)"
 		>
-			<InsertModuleButton index={0} modules={data?.modules ?? []} funcDesc="" />
+			<InsertModuleButton
+				index={0}
+				modules={data?.modules ?? []}
+				funcDesc=""
+				on:new={(e) => {
+					data?.eventHandlers.insert({ modules: data.modules, index: 0, detail: e.detail })
+				}}
+			/>
 		</div>
 	{/if}
 
@@ -67,8 +74,7 @@
 					on:click={() => {
 						data.eventHandlers.insert({
 							modules: data.modules,
-							module: undefined,
-							type: 'move'
+							module: undefined
 						})
 					}}
 					type="button"
diff --git a/frontend/src/lib/components/graph/renderers/nodes/branchAllEndNode.svelte b/frontend/src/lib/components/graph/renderers/nodes/branchAllEndNode.svelte
index ab256a5557..b48b15abfe 100644
--- a/frontend/src/lib/components/graph/renderers/nodes/branchAllEndNode.svelte
+++ b/frontend/src/lib/components/graph/renderers/nodes/branchAllEndNode.svelte
@@ -4,12 +4,14 @@
 	import type { FlowModule } from '$lib/gen/models/FlowModule'
 	import type { GraphEventHandlers } from '../../graphBuilder'
 	import { getStateColor } from '../../util'
+	import type { GraphModuleState } from '../../model'
 
 	export let data: {
 		id: string
 		insertable: boolean
 		modules: FlowModule[]
 		eventHandlers: GraphEventHandlers
+		flowModuleStates: Record<string, GraphModuleState> | undefined
 	}
 </script>
 
@@ -22,6 +24,7 @@
 		selected={false}
 		insertable={false}
 		bgColor={getStateColor(undefined, darkMode)}
+		borderColor={getStateColor(data?.flowModuleStates?.[data?.id]?.type, darkMode)}
 		on:select={(e) => {
 			data?.eventHandlers?.select(e.detail)
 		}}