1 files changed, 5 insertions, 1 deletions
diff --git a/Software/Visual_Studio/Tango.CodeGeneration/ProtoMessageFile.cs b/Software/Visual_Studio/Tango.CodeGeneration/ProtoMessageFile.cs
index 6d771099a..2bcdac8b4 100644
--- a/Software/Visual_Studio/Tango.CodeGeneration/ProtoMessageFile.cs
+++ b/Software/Visual_Studio/Tango.CodeGeneration/ProtoMessageFile.cs
@@ -6,14 +6,18 @@ using System.Threading.Tasks;
 
 namespace Tango.CodeGeneration
 {
-    public class ProtoMessageFile : Class
+    public class ProtoMessageFile : CodeObject
     {
+        public String Name { get; set; }
         public String Package { get; set; }
         public List<String> Imports { get; set; }
 
+        public List<ProtoProperty> Properties { get; set; }
+
         public ProtoMessageFile()
         {
             Imports = new List<string>();
+            Properties = new List<ProtoProperty>();
         }
     }
 }