1 files changed, 10 insertions, 2 deletions
diff --git a/engine/network/index.ts b/engine/network/index.ts
index 1bf95fb..5dc7ece 100644
--- a/engine/network/index.ts
+++ b/engine/network/index.ts
@@ -1,12 +1,20 @@
 export enum MessageType {
   NEW_ENTITIES = 'NEW_ENTITIES',
   REMOVE_ENTITIES = 'REMOVE_ENTITIES',
-  UPDATE_ENTITY = 'UPDATE_ENTITY'
+  UPDATE_ENTITIES = 'UPDATE_ENTITIES',
+  NEW_INPUT = 'NEW_INPUT',
+  REMOVE_INPUT = 'REMOVE_INPUT'
 }
 
 export type EntityAddBody = {
   entityName: string;
-  args: any;
+  id: string;
+  args: Record<string, any>;
+};
+
+export type EntityUpdateBody = {
+  id: string;
+  args: Record<string, any>;
 };
 
 export type Message = {