diff --git a/.idea/.idea.ComfySharp/.idea/vcs.xml b/.idea/.idea.ComfySharp/.idea/vcs.xml
new file mode 100644
index 0000000..94a25f7
--- /dev/null
+++ b/.idea/.idea.ComfySharp/.idea/vcs.xml
@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="$PROJECT_DIR$" vcs="Git" />
+  </component>
+</project>
\ No newline at end of file
diff --git a/ComfySharp/ComfyClient.cs b/ComfySharp/ComfyClient.cs
index 01c6439..809526d 100644
--- a/ComfySharp/ComfyClient.cs
+++ b/ComfySharp/ComfyClient.cs
@@ -81,8 +81,14 @@ public class Node {
     public List<bool> OutputIsList { get; set; }
     [DataMember(Name = "output_name")]
     public List<string> OutputNames { get; set; }
-    
-    
+    [DataMember(Name = "display_name")]
+    public string DisplayName { get; set; }
+    [DataMember(Name = "description")]
+    public string Description { get; set; }
+    [DataMember(Name = "category")]
+    public string Category { get; set; }
+    [DataMember(Name = "output_node")]
+    public bool IsOutputNode { get; set; }
 }
 
 [DataContract]