diff --git a/src/heatmap.rs b/src/heatmap.rs
index 52e5db6..6335659 100644
--- a/src/heatmap.rs
+++ b/src/heatmap.rs
@@ -1,4 +1,3 @@
-use ndarray as nd;
 use image;
 
 pub enum ColorMaps{
diff --git a/src/test.rs b/src/test.rs
index 61d5fb5..fec368c 100644
--- a/src/test.rs
+++ b/src/test.rs
@@ -28,7 +28,7 @@ fn main() {
 
     let mut sw = Stopwatch::start_new();
 
-    let frame = image::open("/home/ruben/Documents/Projecten/2020/rust/lena_orig-s.png");
+    let frame = image::open("test.png");
     
 
     // let vec: Vec<u8> = frame.data().to_vec();
@@ -48,7 +48,7 @@ fn main() {
             let i = ib.as_rgb8().unwrap().clone();
             let hm = Some(heatmap::Heatmap::new(heatmap::ColorMaps::Plasma));
             let image = haar.scan_image(i, &hm).unwrap().dynamic_img;
-            image.save("/home/ruben/Documents/Projecten/2020/rust/lena_orig-output.png");
+            image.save("test-output.png");
 
             // let hm = heatmap::Heatmap::new(heatmap::ColorMaps::NipySpectral);
             // let hm = heatmap::Heatmap::new(heatmap::ColorMaps::TraficLight);