1 files changed, 55 insertions, 49 deletions
diff --git a/examples/oc.py b/examples/oc.py
index f19a2b0..12fd8dd 100644
--- a/examples/oc.py
+++ b/examples/oc.py
@@ -151,53 +151,59 @@ for vname in (
 # ~ v = vars()[vname]
 # ~ v.setDebug()
 
-test = r"""
-/* A factorial program */
-int
-putstr(char *s)
-{
-    while(*s)
-        putchar(*s++);
-}
-
-int
-fac(int n)
-{
-    if (n == 0)
-        return 1;
-    else
-        return n*fac(n-1);
-}
-
-int
-putn(int n)
-{
-    if (9 < n)
-        putn(n / 10);
-    putchar((n%10) + '0');
-}
-
-int
-facpr(int n)
-{
-    putstr("factorial ");
-    putn(n);
-    putstr(" = ");
-    putn(fac(n));
-    putstr("\n");
-}
-
-int
-main()
-{
-    int i;
-    i = 0;
-    if(a() == 1){}
-    while(i < 10)
-        facpr(i++);
-    return 0;
-}
-"""
 
-ast = program.parseString(test, parseAll=True)
-ast.pprint()
+def main():
+    test = r"""
+    /* A factorial program */
+    int
+    putstr(char *s)
+    {
+        while(*s)
+            putchar(*s++);
+    }
+    
+    int
+    fac(int n)
+    {
+        if (n == 0)
+            return 1;
+        else
+            return n*fac(n-1);
+    }
+    
+    int
+    putn(int n)
+    {
+        if (9 < n)
+            putn(n / 10);
+        putchar((n%10) + '0');
+    }
+    
+    int
+    facpr(int n)
+    {
+        putstr("factorial ");
+        putn(n);
+        putstr(" = ");
+        putn(fac(n));
+        putstr("\n");
+    }
+    
+    int
+    main()
+    {
+        int i;
+        i = 0;
+        if(a() == 1){}
+        while(i < 10)
+            facpr(i++);
+        return 0;
+    }
+    """
+
+    ast = program.parseString(test, parseAll=True)
+    ast.pprint()
+
+
+if __name__ == "__main__":
+    main()