diff --git a/Dipolar-Gas-Simulator/+Scripts/run_on_cluster.m b/Dipolar-Gas-Simulator/+Scripts/run_on_cluster.m
index efce9f0..820d96f 100644
--- a/Dipolar-Gas-Simulator/+Scripts/run_on_cluster.m
+++ b/Dipolar-Gas-Simulator/+Scripts/run_on_cluster.m
@@ -8,6 +8,7 @@ OptionsStruct.DipolarAzimuthAngle     = 0;
 OptionsStruct.ScatteringLength        = 98.0676;     % Critical point: 102.515; Triangular phase: 98.0676; Stripe phase: 102.2518; Honeycomb phase: 102.6441
 
 OptionsStruct.TrapFrequencies         = [10, 10, 72.4];
+OptionsStruct.TrapPotentialType       = 'None';
 
 OptionsStruct.NumberOfGridPoints      = [128, 128];
 OptionsStruct.Dimensions              = [7.5, 7.5];  % Critical point: 6.996; Triangular phase: 7.5; Stripe phase: 6.972; Honeycomb phase: 6.239 for both for Atom Number fixed to 1E5
@@ -24,6 +25,8 @@ options                               = Helper.convertstruct2cell(OptionsStruct)
 clear OptionsStruct
 
 solver                                = VariationalSolver2D.DipolarGas(options{:});
+pot                                   = VariationalSolver2D.Potentials(options{:});
+solver.Potential                      = pot.trap();
 
 %-% Run Solver %-%
-[Params, Transf, psi, V, VDk] = solver.run();
\ No newline at end of file
+[Params, Transf, psi, V, VDk]         = solver.run();
\ No newline at end of file
diff --git a/Dipolar-Gas-Simulator/+VariationalSolver2D/@DipolarGas/propagateWavefunction.m b/Dipolar-Gas-Simulator/+VariationalSolver2D/@DipolarGas/propagateWavefunction.m
index bca3b99..2c3bd9a 100644
--- a/Dipolar-Gas-Simulator/+VariationalSolver2D/@DipolarGas/propagateWavefunction.m
+++ b/Dipolar-Gas-Simulator/+VariationalSolver2D/@DipolarGas/propagateWavefunction.m
@@ -26,9 +26,6 @@ function [psi] = propagateWavefunction(this, psi, Params, VParams, Transf, VDk,
     gamma_eff       = Params.gammaQF * (sqrt(2/5)/(pi^(3/4)*VParams.ell^(3/2)));
     Ez              = (0.25*VParams.ell^2) + (0.25*Params.gz*VParams.ell^2);
     
-    pb = Helper.ProgressBar();
-    pb.run('Running simulation in imaginary time: ');
-
     while t_idx < Params.sim_time_cut_off
 
         % kin
@@ -91,9 +88,7 @@ function [psi] = propagateWavefunction(this, psi, Params, VParams, Transf, VDk,
             disp('NaNs encountered!')
             break
         end
-        
         t_idx=t_idx+1;
-        pb.run(100*t_idx/Params.sim_time_cut_off);
     end
     
     % Change in Energy