diff --git a/contrib/docs/templates/driver_builder.py.jinja2 b/contrib/docs/templates/driver_builder.py.jinja2
index 137c23dfa..89f98e7d1 100644
--- a/contrib/docs/templates/driver_builder.py.jinja2
+++ b/contrib/docs/templates/driver_builder.py.jinja2
@@ -32,6 +32,9 @@ dr = (
     .with_modules({{MODULE_NAME}})
     .build()
 )
+# If you have sf-hamilton[visualization] installed, you can see the dataflow graph
+# In a notebook this will show an image, else pass in arguments to save to a file
+# dr.display_all_functions()
 # Execute the dataflow, specifying what you want back. Will return a dictionary.
 result = dr.execute(
     [{{MODULE_NAME}}.CHANGE_ME, ...],  # this specifies what you want back