Updated to use the new URDF model