diff --git a/__init__.py b/__init__.py
index 1f25a97..50083ef 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,11 +1,23 @@
-from .nodes import PlayGame
+from .nodes import LoadDiTModel, LoadVAEModel, LoadTextEncoderModel, LoadGameImage, LoadMouseIcon, GameVideoGenerator, MatrixGameOutput
 
 NODE_CLASS_MAPPINGS = {
-    "PlayGame": PlayGame,
+    "LoadDiTModel": LoadDiTModel,
+    "LoadVAEModel": LoadVAEModel,
+    "LoadTextEncoderModel": LoadTextEncoderModel,
+    "LoadGameImage": LoadGameImage,
+    "LoadMouseIcon": LoadMouseIcon,
+    "GameVideoGenerator": GameVideoGenerator,
+    "MatrixGameOutput": MatrixGameOutput,
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
-    "PlayGame": "Play Game",
+    "LoadDiTModel": "Load DiT Model",
+    "LoadVAEModel": "Load VAE Model",
+    "LoadTextEncoderModel": "Load TextEncoder Model",
+    "LoadGameImage": "Load Game Image",
+    "LoadMouseIcon": "Load Mouse Icon",
+    "GameVideoGenerator": "Game Video Generator",
+    "MatrixGameOutput": "MatrixGame Output",
 } 
 
 __all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']