diff --git a/.github/workflows/build_documentation.yml b/.github/workflows/build_documentation.yml
new file mode 100644
index 0000000..c5eb4bc
--- /dev/null
+++ b/.github/workflows/build_documentation.yml
@@ -0,0 +1,19 @@
+name: Build documentation
+
+on:
+  push:
+    branches:
+      - main
+
+jobs:
+  build:
+    uses: huggingface/doc-builder/.github/workflows/build_main_documentation.yml@main
+    with:
+      commit_sha: ${{ github.sha }}
+      package: diffusion-models-class
+      package_name: diffusion-course
+      path_to_docs: diffusion-models-class/
+      additional_args: --not_python_module
+      convert_notebooks: true
+    secrets:
+      hf_token: ${{ secrets.HF_DOC_BUILD_PUSH }}
diff --git a/.github/workflows/build_pr_documentation.yml b/.github/workflows/build_pr_documentation.yml
new file mode 100644
index 0000000..ad3a01a
--- /dev/null
+++ b/.github/workflows/build_pr_documentation.yml
@@ -0,0 +1,20 @@
+name: Build PR Documentation
+
+on:
+  pull_request:
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
+  cancel-in-progress: true
+
+jobs:
+  build:
+    uses: huggingface/doc-builder/.github/workflows/build_pr_documentation.yml@main
+    with:
+      commit_sha: ${{ github.event.pull_request.head.sha }}
+      pr_number: ${{ github.event.number }}
+      package: diffusion-models-class
+      package_name: diffusion-course
+      path_to_docs: diffusion-models-class/
+      additional_args: --not_python_module
+      convert_notebooks: true
diff --git a/.github/workflows/upload_pr_documentation.yml b/.github/workflows/upload_pr_documentation.yml
new file mode 100644
index 0000000..380b7f3
--- /dev/null
+++ b/.github/workflows/upload_pr_documentation.yml
@@ -0,0 +1,17 @@
+name: Upload PR Documentation
+
+on:
+  workflow_run:
+    workflows: ["Build PR Documentation"]
+    types:
+      - completed
+
+jobs:
+  build:
+    uses: huggingface/doc-builder/.github/workflows/upload_pr_documentation.yml@main
+    with:
+      package_name: diffusion-course
+      hub_base_path: https://moon-ci-docs.huggingface.co/learn
+    secrets:
+      hf_token: ${{ secrets.HF_DOC_BUILD_PUSH }}
+      comment_bot_token: ${{ secrets.COMMENT_BOT_TOKEN }}
\ No newline at end of file
diff --git a/_toctree.yml b/_toctree.yml
index 2b8db7d..95e4fb1 100644
--- a/_toctree.yml
+++ b/_toctree.yml
@@ -40,7 +40,7 @@
       newlocal: unit4/2
       title: DDIM Inversion
     - local: unit4/02_diffusion_for_audio
-      newlocal: unit4/1
+      newlocal: unit4/3
       title: Diffusion for Audio
 
 - title: Events related to the course
diff --git a/unit4/02_diffusion_for_audio.ipynb b/unit4/02_diffusion_for_audio.ipynb
index 9a81abc..03dd8af 100644
--- a/unit4/02_diffusion_for_audio.ipynb
+++ b/unit4/02_diffusion_for_audio.ipynb
@@ -1129,7 +1129,7 @@
     "\n",
     "## Usage\n",
     "\n",
-    "```python\n",
+    "<pre>\n",
     "from IPython.display import Audio\n",
     "from diffusers import DiffusionPipeline\n",
     "\n",
@@ -1137,7 +1137,7 @@
     "output = pipe()\n",
     "display(output.images[0])\n",
     "display(Audio(output.audios[0], rate=pipe.mel.get_sample_rate()))\n",
-    "```\n",
+    "</pre>\n",
     "\"\"\"\n",
     "\n",
     "card = ModelCard(content)\n",