diff --git a/static/koboldai.js b/static/koboldai.js
index 64da7146..75563df2 100644
--- a/static/koboldai.js
+++ b/static/koboldai.js
@@ -151,8 +151,19 @@ const context_menu_actions = {
 	"submit-button": [
 		{label: "Generate", icon: "edit", enabledOn: "ALWAYS", click: function(){}},
 		null,
+		{label: "Generate Forever", icon: "edit_off", enabledOn: "ALWAYS", click: function(){}},
+		{label: "Generate Until EOS", icon: "edit_off", enabledOn: "ALWAYS", click: function(){}},
+		null,
 		{label: "Finish Line", icon: "edit_off", enabledOn: "ALWAYS", click: function(){}},
 		{label: "Finish Sentence", icon: "edit_off", enabledOn: "ALWAYS", click: function(){}},
+	],
+	"undo-button": [
+		{label: "Undo", icon: "undo", enabledOn: "ALWAYS", click: function(){}},
+		null,
+		{label: "Prune Actions", icon: "cut", enabledOn: "ALWAYS", click: function(){}},
+		{label: "Shred", icon: "local_fire_department", enabledOn: "ALWAYS", click: function(){}},
+		null,
+		{label: "Trim Last Sentence", icon: "carpenter", enabledOn: "ALWAYS", click: function(){}},
 	]
 };
 
diff --git a/templates/index_new.html b/templates/index_new.html
index 53bcffd5..2b1c0ddf 100644
--- a/templates/index_new.html
+++ b/templates/index_new.html
@@ -112,7 +112,7 @@
 			</span>
 			<button type="button" class="btn action_button submit var_sync_alt_system_aibusy" system_aibusy=False id="btnsubmit" onclick="storySubmit();" context-menu="submit-button">Submit</button>
 			<button type="button" class="btn action_button submited var_sync_alt_system_aibusy"  system_aibusy=False id="btnsent"><img id="thinking" src="static/thinking.gif" class="force_center" onclick="socket.emit('abort','');"></button>
-			<button type="button" class="btn action_button back var_sync_alt_system_aibusy" system_aibusy=False onclick="storyBack();" aria-label="undo"><span class="material-icons-outlined" style="font-size: 1.4em;">replay</span></button>
+			<button type="button" class="btn action_button back var_sync_alt_system_aibusy" system_aibusy=False onclick="storyBack();" aria-label="undo" context-menu="undo-button"><span class="material-icons-outlined" style="font-size: 1.4em;">replay</span></button>
 			<button type="button" class="btn action_button redo var_sync_alt_system_aibusy" system_aibusy=False onclick="storyRedo();" aria-label="redo"><span class="material-icons-outlined" style="font-size: 1.4em;">arrow_forward</span></button>
 			<button type="button" class="btn action_button retry var_sync_alt_system_aibusy" system_aibusy=False onclick="storyRetry();" aria-label="retry"><span class="material-icons-outlined" style="font-size: 1.4em;">autorenew</span></button>
 		</div>