diff --git a/vlaplex.f90 b/vlaplex.f90
index 992f19a..453b667 100644
--- a/vlaplex.f90
+++ b/vlaplex.f90
@@ -158,7 +158,7 @@ program VlaPlEx
 
   write(*, '(A,ES0.4e3)') 'CFL: ', dt*vf/dr
 
-  nz = 16
+  nz = 2
   ! Allocate vectors
   allocate(f_i(1:nz,1:nr,1:nv), f_i_old(1:nz,1:nr,1:nv))
   allocate(n_i(1:nz,1:nr))
@@ -179,7 +179,7 @@ program VlaPlEx
   n_e     = 0.0_dp
   T_e     = 0.0_dp
   Zave    = 0.0_dp
-  Z_list  = (/(I, I = 1, nz) /)
+  Z_list  = (/ 6.0, 12.0 /)
   phi     = 0.0_dp
   phi_old = 0.0_dp
   E       = 0.0_dp