diff --git a/MOT Capture Process Simulation/@MOTSimulator/initialPositionSampling.m b/MOT Capture Process Simulation/@MOTSimulator/initialPositionSampling.m
index 85af394..197912f 100644
--- a/MOT Capture Process Simulation/@MOTSimulator/initialPositionSampling.m	
+++ b/MOT Capture Process Simulation/@MOTSimulator/initialPositionSampling.m	
@@ -1,7 +1,12 @@
 function ret = initialPositionSampling(this)
-    n = this.NumberOfAtoms;
-    phi = 2 * pi * rand(n,1);
-    rho = this.Beta * 0.5 * this.NozzleLength * sqrt(rand(n,1));
-    ret = [-this.OvenDistance * ones(n,1), rho.*cos(phi), rho.*sin(phi)];
+    switch this.SimulationMode
+        case "2D"
+            n = this.NumberOfAtoms;
+            phi = 2 * pi * rand(n,1);
+            rho = this.Beta * 0.5 * this.NozzleLength * sqrt(rand(n,1));
+            ret = [-this.OvenDistance * ones(n,1), rho.*cos(phi), rho.*sin(phi)];
+        case "3D"
+            % Development In progress
+    end
 end