Quadruped robot

Hi All, I am attempting to achieve a similar functionality to the quadruped example (standalone_examples/api/omni.isaac.quadruped/a1_vision_ros2_standalone.py).
I have imported my robot model (Unitree B2) from Isaac Sim assets store and modified the parameters related to B2. But there are some issues with the model as shown in this video:


Here is the code and model, thanks a lot.
code and model: quadruped - Google Drive