1 files changed, 7 insertions, 4 deletions
diff --git a/src/engine/entities/FunctionBox.ts b/src/engine/entities/FunctionBox.ts
index dac2174..7b70567 100644
--- a/src/engine/entities/FunctionBox.ts
+++ b/src/engine/entities/FunctionBox.ts
@@ -67,7 +67,10 @@ export class FunctionBox extends Entity {
   }
 }
 
-export const makeLambdaTermHighlightComponent = (entity: Entity) => {
+export const makeLambdaTermHighlightComponent = (
+  entity: Entity,
+  text?: string
+) => {
   const onUnhighlight = () => {
     closeModal();
     entity.removeComponent(ComponentNames.Interactable);
@@ -87,9 +90,9 @@ export const makeLambdaTermHighlightComponent = (entity: Entity) => {
         return;
       }
 
-      const code = entity.getComponent<LambdaTerm>(
-        ComponentNames.LambdaTerm
-      )!.code;
+      const code =
+        text ??
+        entity.getComponent<LambdaTerm>(ComponentNames.LambdaTerm)!.code;
       openModal(
         `<div style="text-align:center"><p>${code}</p> <br> <button id="close">Close</button></div>`
       );