diff --git a/Cargo.toml b/Cargo.toml
index 5a4f393..3896e0b 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -18,6 +18,7 @@ approx = "0.5.1"
 candle-core = "0.7.2"
 candle-datasets = "0.7.2"
 candle-nn = "0.7.2"
+candle-transformers = "0.7.2"
 chrono = "0.4.38"
 indicatif = "0.17.8"
 levenberg-marquardt = "0.14.0"
@@ -45,6 +46,7 @@ time = { version = "0.3.36", features = [
     "parsing",
 ], optional = true }
 tokio-test = "0.4.4"
+tracing = "0.1.40"
 yahoo_finance_api = { version = "*", git = "https://github.com/dancixx/yahoo_finance_api.git", rev = "476ebdc", optional = true }
 
 [dev-dependencies]
diff --git a/src/ai.rs b/src/ai.rs
index 65b7647..8646ee4 100644
--- a/src/ai.rs
+++ b/src/ai.rs
@@ -1 +1 @@
-pub mod fou_lstm;
+pub mod fou;
diff --git a/src/ai/fou.rs b/src/ai/fou.rs
new file mode 100644
index 0000000..64822b1
--- /dev/null
+++ b/src/ai/fou.rs
@@ -0,0 +1,4 @@
+pub mod fou_lstm_datasets;
+pub mod fou_lstm_model_1_d;
+pub mod fou_lstm_model_2_d;
+pub mod fou_vae;
diff --git a/src/ai/fou_lstm/datasets.rs b/src/ai/fou/fou_lstm_datasets.rs
similarity index 100%
rename from src/ai/fou_lstm/datasets.rs
rename to src/ai/fou/fou_lstm_datasets.rs
diff --git a/src/ai/fou_lstm/lstm_model_1_d.rs b/src/ai/fou/fou_lstm_model_1_d.rs
similarity index 99%
rename from src/ai/fou_lstm/lstm_model_1_d.rs
rename to src/ai/fou/fou_lstm_model_1_d.rs
index 68d9513..53cd59d 100644
--- a/src/ai/fou_lstm/lstm_model_1_d.rs
+++ b/src/ai/fou/fou_lstm_model_1_d.rs
@@ -8,7 +8,7 @@ use candle_nn::{
 };
 use polars::prelude::*;
 
-use super::datasets::test_vasicek_1_d;
+use super::fou_lstm_datasets::test_vasicek_1_d;
 
 pub struct Model {
   is_train: bool,
diff --git a/src/ai/fou_lstm/lstm_model_2_d.rs b/src/ai/fou/fou_lstm_model_2_d.rs
similarity index 99%
rename from src/ai/fou_lstm/lstm_model_2_d.rs
rename to src/ai/fou/fou_lstm_model_2_d.rs
index 8a8b645..84774e6 100644
--- a/src/ai/fou_lstm/lstm_model_2_d.rs
+++ b/src/ai/fou/fou_lstm_model_2_d.rs
@@ -7,7 +7,7 @@ use candle_nn::{
 };
 use polars::prelude::*;
 
-use super::datasets::test_vasicek_2_d;
+use super::fou_lstm_datasets::test_vasicek_2_d;
 
 pub struct Model {
   is_train: bool,
diff --git a/src/ai/fou_lstm.rs b/src/ai/fou_lstm.rs
deleted file mode 100644
index cc86451..0000000
--- a/src/ai/fou_lstm.rs
+++ /dev/null
@@ -1,3 +0,0 @@
-pub mod datasets;
-pub mod lstm_model_1_d;
-pub mod lstm_model_2_d;