diff --git a/src/compute/mod.rs b/src/compute/mod.rs
index b24f638..8986bdc 100644
--- a/src/compute/mod.rs
+++ b/src/compute/mod.rs
@@ -1,3 +1,4 @@
-pub mod activations;
 
-pub mod models;
\ No newline at end of file
+pub mod models;
+
+pub mod stats;
\ No newline at end of file
diff --git a/src/compute/activations.rs b/src/compute/models/activations.rs
similarity index 98%
rename from src/compute/activations.rs
rename to src/compute/models/activations.rs
index 661fe52..e9f528a 100644
--- a/src/compute/activations.rs
+++ b/src/compute/models/activations.rs
@@ -130,6 +130,6 @@ mod tests {
     fn test_dleaky_relu_edge_case() {
         let x = Matrix::from_vec(vec![-1e-10, 0.0, 1e10], 3, 1);
         let expected = Matrix::from_vec(vec![0.01, 0.01, 1.0], 3, 1);
-        assert_eq!(dleaky_relu(&x), expected);
+        assert_eq!(dleaky_relu(&x), expectewd);
     }
 }
diff --git a/src/compute/models/dense_nn.rs b/src/compute/models/dense_nn.rs
index 7edfaab..b795b93 100644
--- a/src/compute/models/dense_nn.rs
+++ b/src/compute/models/dense_nn.rs
@@ -1,4 +1,4 @@
-use crate::compute::activations::{drelu, relu, sigmoid};
+use crate::compute::models::activations::{drelu, relu, sigmoid};
 use crate::matrix::{Matrix, SeriesOps};
 use rand::prelude::*;
 
diff --git a/src/compute/models/logreg.rs b/src/compute/models/logreg.rs
index 8821ee4..ac224aa 100644
--- a/src/compute/models/logreg.rs
+++ b/src/compute/models/logreg.rs
@@ -1,4 +1,4 @@
-use crate::compute::activations::sigmoid;
+use crate::compute::models::activations::sigmoid;
 use crate::matrix::{Matrix, SeriesOps};
 
 pub struct LogReg {
diff --git a/src/compute/models/mod.rs b/src/compute/models/mod.rs
index c617f08..051d523 100644
--- a/src/compute/models/mod.rs
+++ b/src/compute/models/mod.rs
@@ -3,4 +3,5 @@ pub mod logreg;
 pub mod dense_nn;
 pub mod k_means;    
 pub mod pca;
-pub mod gaussian_nb;
\ No newline at end of file
+pub mod gaussian_nb;
+pub mod activations;