code: def new_geom(geom_type, size, pos=(0, 0, 0), rgba=RED, group=0, **kwargs):
      
      kwargs["type"] = str(geom_type)
      kwargs["size"] = array_to_string(size)
      kwargs["rgba"] = array_to_string(rgba)
      kwargs["group"] = str(group)
      kwargs["pos"] = array_to_string(pos)
      element = ET.Element("geom", attrib=kwargs)
      return element
docs: Creates a geom element with attributes specified by @**kwargs.
  
      Args:
          geom_type (str): type of the geom.
              see all types here: http://mujoco.org/book/modeling.html#geom
          size: geom size parameters.
          pos: 3d position of the geom frame.
          rgba: color and transparency. Defaults to solid red.
          group: the integrer group that the geom belongs to. useful for
              separating visual and physical elements.