1 files changed, 8 insertions, 4 deletions

diff --git a/chunk.d b/chunk.d
index 87c9fb6..c512f14 100644
--- a/chunk.d
+++ b/chunk.d
@@ -6,6 +6,7 @@ import parser;
 
 enum ObjType {
     FUNCTION,
+    SCRIPT,
 }
 
 abstract class Obj {
@@ -14,19 +15,22 @@ abstract class Obj {
 
 class Function : Obj {
     Chunk chunk;
+    int arity;
     string name;
+    ObjType type;
 
-    this() {
-        this.type = ObjType.FUNCTION;
+    this(ObjType type) {
+        this.type = type;
         this.chunk = new Chunk();
+        this.arity = 0;
         this.name = "";
     }
 
     override string toString() {
-        if (name == "") {
+        if (type == ObjType.SCRIPT) {
             return "<neb>";
         } else {
-            return name;
+            return format("<fn %s>", name);
         }
     }
 }