diff --git a/XZ_technique.jl b/XZ_technique.jl
index 284ca69..aee4fac 100644
--- a/XZ_technique.jl
+++ b/XZ_technique.jl
@@ -2,21 +2,22 @@ using LinearAlgebra, Plots
 include("ho_basis.jl")
 include("p_space.jl")
 
+ω = 0.25
 μ = 0.5
 l = 0
 V1 = -8.6
 R1 = 1.75
 V2 = 3.4
 R2 = 3.2
-n_max = 30
+n_max = 20
 
 ns = collect(0:n_max)
 ls = fill(l, n_max + 1)
 
-T = 1/μ * sp_T_matrix(ns, ls)
-V = V1 .* V_Gaussian.(R1, l, ns, transpose(ns)) + V2 .* V_Gaussian.(R2, l, ns, transpose(ns))
+T = 1/μ * sp_T_matrix(ns, ls; ω=ω)
+V = V1 .* V_Gaussian.(R1, l, ns, transpose(ns); ω=ω) + V2 .* V_Gaussian.(R2, l, ns, transpose(ns); ω=ω)
 
-n_EC = 16
+n_EC = 8
 train_cs = (2.2 - 0.7im) .+ (0.1 .* randn(n_EC)) .+ (0.1im * randn(n_EC))
 target_cs = range(2.0, 0.8, 6)