diff --git a/docs/src/tensor_prod.md b/docs/src/tensor_prod.md
index e6518e17..b6e56f51 100644
--- a/docs/src/tensor_prod.md
+++ b/docs/src/tensor_prod.md
@@ -14,14 +14,16 @@ using Plots
 default()
 ```
 
-# Generating Data and Plotting
+Generating Data and Plotting
 
 ```@example tensor
 function tensor_product_function(x, a)
     return prod(cos.(a * π * xi) for xi in x)
 end
+```
 
-# Generate training and test data
+Generate training and test data
+```@example tensor
 function generate_data(n, lb, ub, a)
     x_train = sample(n, lb, ub, SobolSample())
     y_train = tensor_product_function(x_train, a)
@@ -31,16 +33,20 @@ function generate_data(n, lb, ub, a)
     
     return x_train, y_train, x_test, y_test
 end
+```
 
-# Visualize training data and the true function
+Visualize training data and the true function
+```example tensor
 function plot_data_and_true_function(x_train, y_train, x_test, y_test, a, lb, ub)
     xs = range(lb, ub, length=1000)
     plot(xs, tensor_product_function.(xs, a), label="True Function", legend=:top)
     scatter!(x_train, repeat([y_train], length(x_train)), label="Training Points", xlims=(lb,ub), ylims=(-1,1))
     scatter!(x_test, repeat([y_test], length(x_test)), label="Test Points")
 end
+```
 
-# Generate data and plot
+Generate data and plot
+```@example tensor
 n = 30
 lb = -5.0
 ub = 5.0
@@ -50,25 +56,27 @@ x_train, y_train, x_test, y_test = generate_data(n, lb, ub, a)
 plot_data_and_true_function(x_train, y_train, x_test, y_test, a, lb, ub)
 ```
 
-# Training various Surrogates
-Now let's train various surrogate models and evaluate their performance on the test data
+Training various Surrogates
 
-```
-# Train different surrogate models
+```@example tensor
 function train_surrogates(x_train, y_train, lb, ub, alpha=2.0, n=6)
     loba = LobachevskySurrogate(x_train, y_train, lb, ub, alpha=alpha, n=n)
     krig = Kriging(x_train, y_train, lb, ub)
     return loba, krig
 end
+```
 
-# Evaluate and compare surrogate model performances
+Evaluate and compare surrogate model performances
+```@example tensor
 function evaluate_surrogates(loba, krig, x_test)
     loba_pred = loba.(x_test)
     krig_pred = krig.(x_test)
     return loba_pred, krig_pred
 end
+```
 
-# Plot surrogate predictions against the true function
+Plot surrogate predictions against the true function
+```@example tensor
 function plot_surrogate_predictions(loba_pred, krig_pred, x_test, y_test, a, lb, ub)
     xs = collect(x_test)  # Convert x_test to an array
     plot(xs, tensor_product_function.(xs, a), label="True Function", legend=:top)
@@ -76,28 +84,34 @@ function plot_surrogate_predictions(loba_pred, krig_pred, x_test, y_test, a, lb,
     plot!(collect(x_test), krig_pred, seriestype=:scatter, label="Kriging")
     plot!(collect(x_test), fill(y_test, length(x_test)), seriestype=:scatter, label="Sampled points")  # Use fill to create an array of the same length as x_test
 end
+```
 
-# Train surrogates and evaluate their performance
+Train surrogates and evaluate their performance
+```@example tensor
 lb, ub = minimum(x_train), maximum(x_train)
 loba, krig = train_surrogates(x_train, y_train, lb, ub)
 loba_pred, krig_pred = evaluate_surrogates(loba, krig, x_test)
+```
 
-# Plotting Results
+Plotting Results
+```@example tensor
 plot_surrogate_predictions(loba_pred, krig_pred, x_test, y_test, 2.0, lb, ub)
 ```
 
-# Reporting the best Surrogate Model
+Reporting the best Surrogate Model
 To determine the best surrogate, you can compare their accuracy and performance metrics on the test data. For instance, you can calculate and compare the mean squared error (MSE) or any other relevant metric
 
-```
+```@example tensor
 using Statistics
 
 # Evaluate performance metrics
 function calculate_performance_metrics(pred, true_vals)
     return mean((pred .- true_vals).^2)
 end
+```
 
-# Compare surrogate model performances
+Compare surrogate model performances
+```@example tensor
 mse_loba = calculate_performance_metrics(loba_pred, y_test)
 mse_krig = calculate_performance_metrics(krig_pred, y_test)