diff --git a/setup.py b/setup.py
index 54d8cac..8c7b58b 100644
--- a/setup.py
+++ b/setup.py
@@ -17,5 +17,6 @@ setup(name='unitree_sdk2py',
             "cyclonedds==0.10.2",
             "numpy",
             "opencv-python",
+            "pygame",
       ],
-      )
\ No newline at end of file
+      )
diff --git a/unitree_sdk2py/b2/__init__.py b/unitree_sdk2py/b2/__init__.py
new file mode 100644
index 0000000..e69de29