code: def add_class(node: minidom.Node, class_name: str):
      if "class" not in node.attributes:
          node.attributes["class"] = class_name
          return
  
      node.attributes["class"] = (
          node.attributes["class"].value + f" {class_name}"
      )
  
file_path: annotated_deep_learning_paper_implementations/utils/diagrams.py
imports: import shutil
  from pathlib import Path
  from typing import List
  from xml.dom import minidom
  from labml import monit
label_desc: Use image node_id 2 for calling a global function with example usage: add_class(node, class_name) without return types