diff --git a/scripts/run_simulation.jl b/scripts/run_simulation.jl
index 4949395..226ccf6 100644
--- a/scripts/run_simulation.jl
+++ b/scripts/run_simulation.jl
@@ -1,6 +1,8 @@
 using GLMakie, Observables
 include("../src/constants.jl")
 using .Constants
+include("../src/visualization.jl")
+using .Visualization
 
 # # Gray Scott Model Parameters
 # Parameters and initial conditions
@@ -19,7 +21,7 @@ stepsize = Observable{Int}(30)
 
 
 
-function update_params!(params_obs, u, v, feed, kill)
+function update_params!(params_obs::Observable, u, v, feed, kill)
     old = params_obs[]
     params_obs[] = GSParams(old.N, old.dx, u, v, feed, kill)
 end
@@ -39,6 +41,7 @@ V[center-radius:center+radius, center-radius:center+radius] .= 0.25
 # Observable holding current U for heatmap
 heat_obs = Observable(U)
 
+
 function laplacian5(f)
     h2 = dx^2
     left = f[2:end-1, 1:end-2]