diff --git a/src/surface_probability_measure.rs b/src/surface_probability_measure.rs
index 759cd52..25c46a7 100644
--- a/src/surface_probability_measure.rs
+++ b/src/surface_probability_measure.rs
@@ -1,18 +1,12 @@
-use std::fs::File;
-use std::path::Path;
-use bevy::render::render_resource::AsBindGroupShaderType;
-use kd_tree::KdTree;
 use rand::rngs::SmallRng;
 use rand::{Rng, SeedableRng};
 use crate::system::model::DLASystem;
-use crate::system::{Position, Storage};
-use crate::system::spaces::continuous::{ContinuousSticker, ContinuousStorage, ContinuousWalker, P3};
-use crate::system::spaces::hexagonal::HexPosition;
-use crate::system::spaces::square_grid::{Grid2D, Grid3D};
+use crate::system::{Storage};
+use crate::system::spaces::square_grid::Grid2D;
 use crate::system::spaces::VectorStorage;
-use crate::system::spawner::{Spawner, UniformSpawner};
+use crate::system::spawner::UniformSpawner;
 use crate::system::sticker::{ProbabilisticSticking, SimpleSticking, Sticker};
-use crate::system::walker::{LocalRandomWalker, Walker};
+use crate::system::walker::LocalRandomWalker;
 
 struct LoggerSticker {
     inner: SimpleSticking
@@ -42,7 +36,7 @@ impl Storage<Grid2D> for ReadOnlyVectorStorage {
     }
 }
 
-pub fn surface_probability_measure(seed: u64, max_particles: usize, stick_probability: f32) -> DLASystem<SmallRng, HexPosition, VectorStorage, LocalRandomWalker, UniformSpawner, ProbabilisticSticking> {
+pub fn surface_probability_measure(seed: u64, max_particles: usize, stick_probability: f32) -> DLASystem<SmallRng, Grid2D, VectorStorage, LocalRandomWalker, UniformSpawner, ProbabilisticSticking> {
     DLASystem::new(
         SmallRng::seed_from_u64(seed),
         VectorStorage::new(1600, 3),