1 files changed, 10 insertions, 0 deletions
diff --git a/src/main/java/org/zwobble/couscous/ast/ClassNode.java b/src/main/java/org/zwobble/couscous/ast/ClassNode.java
index 3545084..8b54be6 100644
--- a/src/main/java/org/zwobble/couscous/ast/ClassNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/ClassNode.java
@@ -1,6 +1,7 @@
 package org.zwobble.couscous.ast;
 
 import org.zwobble.couscous.ast.visitors.NodeMapper;
+import org.zwobble.couscous.ast.visitors.NodeTransformer;
 
 import java.util.List;
 import java.util.Set;
@@ -66,6 +67,15 @@ public class ClassNode implements Node {
         return visitor.visit(this);
     }
 
+    public ClassNode transform(NodeTransformer transformer) {
+        return new ClassNode(
+            transformer.transform(name),
+            transformer.transformTypes(superTypes),
+            transformer.transformFields(fields),
+            transformer.visit(constructor),
+            transformer.transformMethods(methods));
+    }
+
     @Override
     public String toString() {
         return "ClassNode(" +