diff --git a/src/extension.ts b/src/extension.ts
index 75ce129..275f318 100644
--- a/src/extension.ts
+++ b/src/extension.ts
@@ -8,6 +8,36 @@ import { register as registerCellTagsView } from './cellTagsTreeDataProvider';
 export function activate(context: vscode.ExtensionContext) {
 	registerCellTags(context);
 	registerCellTagsView(context);
+	
+	// Update context when the active editor or selection changes
+	vscode.window.onDidChangeActiveNotebookEditor(updateContext);
+	vscode.window.onDidChangeNotebookEditorSelection(updateContext);
+
+	// Initialize the status bar
+	updateContext();
 }
 
+function updateContext() {
+    const editor = vscode.window.activeNotebookEditor;
+    if (!editor) {
+        vscode.commands.executeCommand('setContext', 'jupyter-cell-tags.singleCellSelected', false);
+        vscode.commands.executeCommand('setContext', 'jupyter-cell-tags.multipleCellsSelected', false);
+        console.log('No active notebook editor');
+        debugSelectedCellsStatusBarItem.hide();
+        return;
+    }
+
+    const selectionCount = editor.selections.length;
+    vscode.commands.executeCommand('setContext', 'jupyter-cell-tags.singleCellSelected', selectionCount === 1);
+    vscode.commands.executeCommand('setContext', 'jupyter-cell-tags.multipleCellsSelected', selectionCount > 1);
+
+    console.log(`Selection count: ${selectionCount}`);
+    console.log(`Single cell selected: ${selectionCount === 1}`);
+    console.log(`Multiple cells selected: ${selectionCount > 1}`);
+
+    debugSelectedCellsStatusBarItem.text = `$(notebook) ${selectionCount} Cell(s) Selected`;
+    debugSelectedCellsStatusBarItem.show();
+}
+
+
 export function deactivate() {}