diff --git a/package.json b/package.json
index f2635956..39b431a1 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "puppeteer-recorder",
-  "version": "0.3.0",
+  "version": "0.4.0",
   "description": "A Chrome extension for recording browser interaction and generating Puppeteer scripts",
   "main": "index.js",
   "scripts": {
diff --git a/src/manifest.json b/src/manifest.json
index a6701b6f..94372c84 100644
--- a/src/manifest.json
+++ b/src/manifest.json
@@ -1,6 +1,6 @@
 {
   "name": "Puppeteer Recorder",
-  "version": "0.3.0",
+  "version": "0.4.0",
   "manifest_version": 2,
   "description": "A Chrome extension for recording browser interaction and generating Puppeteer scripts",
   "permissions": [
diff --git a/src/popup/components/__tests__/__snapshots__/App.spec.js.snap b/src/popup/components/__tests__/__snapshots__/App.spec.js.snap
index 84d40a0c..8347f29d 100644
--- a/src/popup/components/__tests__/__snapshots__/App.spec.js.snap
+++ b/src/popup/components/__tests__/__snapshots__/App.spec.js.snap
@@ -17,7 +17,7 @@ exports[`App.vue it has the correct pristine / empty state 1`] = `
         class="text-muted"
       >
         <small>
-          0.3.0
+          0.4.0
         </small>
       </span>
     </a>
diff --git a/src/popup/components/__tests__/__snapshots__/ResultsTab.spec.js.snap b/src/popup/components/__tests__/__snapshots__/ResultsTab.spec.js.snap
index 5b818ce4..800b51bb 100644
--- a/src/popup/components/__tests__/__snapshots__/ResultsTab.spec.js.snap
+++ b/src/popup/components/__tests__/__snapshots__/ResultsTab.spec.js.snap
@@ -7,14 +7,18 @@ exports[`RecordingTab.vue it has the correct pristine / empty state 1`] = `
   <div
     class="content"
   >
-    <pre>
-            
-      <code>
-        No code yet...
-      </code>
-      
+    <div
+      class="generated-code"
+    >
+      <pre>
+              
+        <code>
+          No code yet...
+        </code>
+        
     
-    </pre>
+      </pre>
+    </div>
   </div>
 </div>
 `;
@@ -26,11 +30,15 @@ exports[`RecordingTab.vue it show a code box when there is code 1`] = `
   <div
     class="content"
   >
-    <pre>
-      <code
-        class="javascript"
-      />
-    </pre>
+    <div
+      class="generated-code"
+    >
+      <pre>
+        <code
+          class="javascript"
+        />
+      </pre>
+    </div>
   </div>
 </div>
 `;