docs: def _set_body(self, body):
          """Set the main body for this control flow structure."""
          assert isinstance(body, CodeStatement)
          if isinstance(body, CodeBlock):
              self.body = body
          else:
              self.body._add(body)