diff --git a/06_gpu_and_ml/long-training.py b/06_gpu_and_ml/long-training.py
index 2802a44ce..7e0699c8f 100644
--- a/06_gpu_and_ml/long-training.py
+++ b/06_gpu_and_ml/long-training.py
@@ -130,8 +130,15 @@ def train(experiment):
 
 # Note that the more common way to wrap functions
 # is by putting `@app.function` as a decorator on the function's definition,
-# but we've split the two steps out in this example to make the separation of concerns clearer.
+# like this:
+
+# ```python
+# @app.function(...)
+# def train(...):
+#     ...
+# ```
 
+# but we've split the two steps out in this example to make the separation of concerns clearer.
 
 # ## Kicking off interruptible training