1 files changed, 15 insertions, 2 deletions
diff --git a/src/engine/entities/FunctionBox.ts b/src/engine/entities/FunctionBox.ts
index b7015f2..e51eb2b 100644
--- a/src/engine/entities/FunctionBox.ts
+++ b/src/engine/entities/FunctionBox.ts
@@ -1,4 +1,10 @@
-import { IMAGES, SPRITE_SPECS, SpriteSpec, Sprites } from "../config";
+import {
+  IMAGES,
+  Miscellaneous,
+  SPRITE_SPECS,
+  SpriteSpec,
+  Sprites,
+} from "../config";
 import { Entity, EntityNames } from ".";
 import {
   BoundingBox,
@@ -77,7 +83,14 @@ export class FunctionBox extends Entity {
         return;
       }
       modalOpen = true;
-      openModal(this.code);
+      openModal(
+        `<div style="text-align:center"><p>${this.code}</p> <br> <button id="close">Close</button></div>`,
+      );
+
+      document.getElementById("close")!.addEventListener("click", () => {
+        closeModal();
+        document.getElementById(Miscellaneous.CANVAS_ID)!.focus();
+      });
     };
 
     this.addComponent(new Interactable(interaction));