From d530483d4a5eeb7a099d1a3224db61e8bdfc64db Mon Sep 17 00:00:00 2001
From: Farid Yagubbayli <faridyagubbayli@gmail.com>
Date: Sun, 5 Nov 2023 18:18:48 +0100
Subject: [PATCH] Add Colab badge

---
 examples/example_na_controlling_the_pml/README.md      |  4 ++++
 .../example_na_controlling_the_pml.ipynb               | 10 ++++++++++
 2 files changed, 14 insertions(+)

diff --git a/examples/example_na_controlling_the_pml/README.md b/examples/example_na_controlling_the_pml/README.md
index 2d61f699..6d51a303 100644
--- a/examples/example_na_controlling_the_pml/README.md
+++ b/examples/example_na_controlling_the_pml/README.md
@@ -1,5 +1,9 @@
 # Using An Ultrasound Transducer As A Sensor Example
 
+<a target="_blank" href="https://colab.research.google.com/github/waltsims/k-wave-python/blob/master/examples/example_na_controlling_the_pml/example_na_controlling_the_pml.ipynb">
+  <img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/>
+</a>
+
 # Running the example
 
 Please have a look at the `example_na_controlling_the_pml.ipynb` notebook file to see the example in action.
diff --git a/examples/example_na_controlling_the_pml/example_na_controlling_the_pml.ipynb b/examples/example_na_controlling_the_pml/example_na_controlling_the_pml.ipynb
index eeeb1ff9..9982ea7b 100644
--- a/examples/example_na_controlling_the_pml/example_na_controlling_the_pml.ipynb
+++ b/examples/example_na_controlling_the_pml/example_na_controlling_the_pml.ipynb
@@ -18,6 +18,16 @@
     "The example is based on the original kWave example in Matlab. A few parts were adapted to make it compatible with `k-wave-python`."
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Install k-wave-python if not installed yet (e.g. running in Google Colab)\n",
+    "#!pip install git+https://github.com/waltsims/k-wave-python"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 1,