diff --git a/Project.toml b/Project.toml
index 889d474..dfa6fc0 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,7 +1,7 @@
 name = "WaveletsExt"
 uuid = "8f464e1e-25db-479f-b0a5-b7680379e03f"
 authors = ["Zeng Fung Liew <lzfliew@ucdavis.edu>", "Shozen Dan <shodan@ucdavis.edu>"]
-version = "0.1.16"
+version = "0.1.17"
 
 [deps]
 AverageShiftedHistograms = "77b51b56-6f8f-5c3a-9cb4-d71f9594ea6e"
diff --git a/paper/paper.md b/paper/paper.md
index 3e9bb68..cd0a391 100644
--- a/paper/paper.md
+++ b/paper/paper.md
@@ -71,12 +71,18 @@ xw = wpdall(x, wt, 6)
 
 # ----- Joint Best Basis (JBB)
 tree = bestbasistree(xw, JBB())
-p1 = plot_tfbdry(tree, node_color=:green, line_color=:black, background_color=:white) |> 
+p1 = plot_tfbdry(tree, 
+                 node_color=:green, 
+                 line_color=:black, 
+                 background_color=:white) |> 
      p -> plot!(p, title="JBB")
 
 # ----- Least Statistically Dependent Basis (LSDB)
 tree = bestbasistree(xw, LSDB())
-p2 = plot_tfbdry(tree, node_color=:green, line_color=:black, background_color=:white) |> 
+p2 = plot_tfbdry(tree, 
+                 node_color=:green, 
+                 line_color=:black, 
+                 background_color=:white) |> 
      p -> plot!(p, title="LSDB")
 
 # Combine and save plot
@@ -159,7 +165,10 @@ ldb = LocalDiscriminantBasis(
 X̂ = fit_transform(ldb, X, y)
 
 # Plot the best basis for feature extraction
-p2 = plot_tfbdry(ldb.tree, node_color=:green, line_color=:black, background_color=:white)
+p2 = plot_tfbdry(ldb.tree, 
+                 node_color=:green, 
+                 line_color=:black, 
+                 background_color=:white)
 plot!(p2, title="Basis Selection using LDB")
 
 # Combine and save plot