From 4d4209656dde3a9d7ccee9f70d35f1e23f452318 Mon Sep 17 00:00:00 2001
From: NikolajDanger <nikolaj@ingemanngade.net>
Date: Fri, 12 Apr 2024 14:42:21 +0200
Subject: [PATCH] :sparkles:

---
 examples/pol_not_calc.plthy | 28 ++++++++++++++--------------
 1 file changed, 14 insertions(+), 14 deletions(-)

diff --git a/examples/pol_not_calc.plthy b/examples/pol_not_calc.plthy
index 7570292..aa323b0 100644
--- a/examples/pol_not_calc.plthy
+++ b/examples/pol_not_calc.plthy
@@ -36,25 +36,25 @@ define calc<3> as [
 ]|
 
 define unfold<1> as [
+    // In case there are 0, 1 or 2 arguments
     do return 'error' if do len<argument #1;> = 0|
-
-    set {} -> middle|
-    set 1 -> i|
     do [
-        set variable middle + {argument #1{variable i};} -> middle|
-        set variable i + 1 -> i|
-    ] until variable i = do len<argument #1;> - 1|
+        do return do int<argument #1{0};> if do "is_number"<argument #1;>|
+        return 'error'|
+    ] if do len<argument #1;> = 1|
+    do return 'error' if do len<argument #1;> = 2|
 
-    do return 'error' if do len<variable middle;> = 0|
+    set argument #1{0} -> op|
 
+    set {} -> n1l|
+    set 1 -> n|
+    set 0 -> i|
     do [
-        set variable middle{0} -> middle|
-        do set do int<variable middle;> -> n1 else return 'error' if do "is_number"<variable middle;>|
-    ] else set do "unfold"<variable middle;> -> n1 if do len<variable middle;> = 1|
-
-    do set do int<argument #1{do len<argument #1;>-1};> -> n2 else return 'error' if do "is_number"<argument #1{do len<argument #1;>-1};>|
-
-    return do "calc"<argument #1{0}; variable n1; variable n2;>|
+        set variable i+1 -> i|
+        set variable n1l + {argument #1{variable i};} -> n1l|
+        do skip else set variable n+2 -> n if do "is_number"<argument #1{variable i};>|
+    ] until variable n = variable i|
+    do print<variable n1l;>|
 ]|
 
 set do input<':';> -> x|