2 files changed, 5 insertions, 1 deletions
diff --git a/extension/src/diffs.ts b/extension/src/diffs.ts
index 98b8753a..b7acd109 100644
--- a/extension/src/diffs.ts
+++ b/extension/src/diffs.ts
@@ -62,7 +62,7 @@ class DiffManager {
   }
 
   private escapeFilepath(filepath: string): string {
-    return filepath.replace(/\\/g, "_").replace(/\//g, "_");
+    return encodeURIComponent(filepath);
   }
 
   private remoteTmpDir: string = "/tmp/continue";
diff --git a/extension/src/lang-server/codeLens.ts b/extension/src/lang-server/codeLens.ts
index ba80e557..ec03f73e 100644
--- a/extension/src/lang-server/codeLens.ts
+++ b/extension/src/lang-server/codeLens.ts
@@ -69,6 +69,10 @@ class DiffViewerCodeLensProvider implements vscode.CodeLensProvider {
           title: `Reject All ❌ (${getMetaKeyLabel()}⇧⌫)`,
           command: "continue.rejectDiff",
           arguments: [document.uri.fsPath],
+        }),
+        new vscode.CodeLens(range, {
+          title: `Further Edit ✏️ (${getMetaKeyLabel()}⇧M)`,
+          command: "continue.focusContinueInputWithEdit",
         })
       );
       return codeLenses;