diff --git a/.clang-tidy b/.clang-tidy
index ef88702..210df1d 100644
--- a/.clang-tidy
+++ b/.clang-tidy
@@ -10,7 +10,7 @@ Checks: '-*,
          -modernize-use-trailing-return-type,
          -bugprone-exception-escape'
 
-HeaderFilterRegex: 'src/*'
+HeaderFilterRegex: 'bayesnet/*'
 AnalyzeTemporaryDtors: false
 WarningsAsErrors: ''
 FormatStyle: file
diff --git a/CHANGELOG.md b/CHANGELOG.md
index 7c9dee2..be54df2 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -5,7 +5,7 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
-## [unpublished]
+## [unreleased]
 
 ### Added
 
diff --git a/CMakeLists.txt b/CMakeLists.txt
index 66eddde..fedd2e7 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -66,7 +66,7 @@ add_git_submodule("lib/json")
 # --------------
 add_subdirectory(config)
 add_subdirectory(lib/Files)
-add_subdirectory(src)
+add_subdirectory(bayesnet)
 
 # Testing
 # -------
@@ -83,5 +83,5 @@ install(TARGETS BayesNet
         ARCHIVE DESTINATION lib
         LIBRARY DESTINATION lib
         CONFIGURATIONS Release)
-install(DIRECTORY src/ DESTINATION include/bayesnet FILES_MATCHING CONFIGURATIONS Release PATTERN "*.h")
+install(DIRECTORY bayesnet/ DESTINATION include/bayesnet FILES_MATCHING CONFIGURATIONS Release PATTERN "*.h")
 install(FILES ${CMAKE_BINARY_DIR}/configured_files/include/config.h DESTINATION include/bayesnet CONFIGURATIONS Release)
\ No newline at end of file
diff --git a/Makefile b/Makefile
index 98c8b22..e7ce446 100644
--- a/Makefile
+++ b/Makefile
@@ -69,6 +69,7 @@ release: ## Build a Release version of the project
 fname = "tests/data/iris.arff"
 sample: ## Build sample
 	@echo ">>> Building Sample...";
+	@if [ -d ./sample/build ]; then rm -rf ./sample/build; fi
 	@cd sample && cmake -B build -S . && cmake --build build -t bayesnet_sample
 	sample/build/bayesnet_sample $(fname)
 	@echo ">>> Done";	
diff --git a/src/BaseClassifier.h b/bayesnet/BaseClassifier.h
similarity index 100%
rename from src/BaseClassifier.h
rename to bayesnet/BaseClassifier.h
index 8e5631c..69f4b29 100644
--- a/src/BaseClassifier.h
+++ b/bayesnet/BaseClassifier.h
@@ -1,8 +1,8 @@
 #ifndef BASE_H
 #define BASE_H
+#include <vector>
 #include <torch/torch.h>
 #include <nlohmann/json.hpp>
-#include <vector>
 namespace bayesnet {
     enum status_t { NORMAL, WARNING, ERROR };
     class BaseClassifier {
diff --git a/src/CMakeLists.txt b/bayesnet/CMakeLists.txt
similarity index 91%
rename from src/CMakeLists.txt
rename to bayesnet/CMakeLists.txt
index 024362c..c1aebe9 100644
--- a/src/CMakeLists.txt
+++ b/bayesnet/CMakeLists.txt
@@ -3,7 +3,7 @@ include_directories(
     ${BayesNet_SOURCE_DIR}/lib/Files
     ${BayesNet_SOURCE_DIR}/lib/folding
     ${BayesNet_SOURCE_DIR}/lib/json/include
-    ${BayesNet_SOURCE_DIR}/src
+    ${BayesNet_SOURCE_DIR}
     ${CMAKE_BINARY_DIR}/configured_files/include
 )
 
diff --git a/src/classifiers/Classifier.cc b/bayesnet/classifiers/Classifier.cc
similarity index 99%
rename from src/classifiers/Classifier.cc
rename to bayesnet/classifiers/Classifier.cc
index 10a5b2d..03c8b57 100644
--- a/src/classifiers/Classifier.cc
+++ b/bayesnet/classifiers/Classifier.cc
@@ -1,5 +1,5 @@
+#include "bayesnet/utils/bayesnetUtils.h"
 #include "Classifier.h"
-#include "utils/bayesnetUtils.h"
 
 namespace bayesnet {
     Classifier::Classifier(Network model) : model(model), m(0), n(0), metrics(Metrics()), fitted(false) {}
diff --git a/src/classifiers/Classifier.h b/bayesnet/classifiers/Classifier.h
similarity index 96%
rename from src/classifiers/Classifier.h
rename to bayesnet/classifiers/Classifier.h
index 103ad7c..c7685a2 100644
--- a/src/classifiers/Classifier.h
+++ b/bayesnet/classifiers/Classifier.h
@@ -1,9 +1,9 @@
 #ifndef CLASSIFIER_H
 #define CLASSIFIER_H
 #include <torch/torch.h>
-#include "BaseClassifier.h"
-#include "bayesian_network/Network.h"
-#include "utils/BayesMetrics.h"
+#include "bayesnet/utils/BayesMetrics.h"
+#include "bayesnet/network/Network.h"
+#include "bayesnet/BaseClassifier.h"
 
 namespace bayesnet {
     class Classifier : public BaseClassifier {
diff --git a/src/classifiers/KDB.cc b/bayesnet/classifiers/KDB.cc
similarity index 100%
rename from src/classifiers/KDB.cc
rename to bayesnet/classifiers/KDB.cc
diff --git a/src/classifiers/KDB.h b/bayesnet/classifiers/KDB.h
similarity index 93%
rename from src/classifiers/KDB.h
rename to bayesnet/classifiers/KDB.h
index 166803a..9478475 100644
--- a/src/classifiers/KDB.h
+++ b/bayesnet/classifiers/KDB.h
@@ -1,8 +1,8 @@
 #ifndef KDB_H
 #define KDB_H
 #include <torch/torch.h>
+#include "bayesnet/utils/bayesnetUtils.h"
 #include "Classifier.h"
-#include "utils/bayesnetUtils.h"
 namespace bayesnet {
     class KDB : public Classifier {
     private:
diff --git a/src/classifiers/KDBLd.cc b/bayesnet/classifiers/KDBLd.cc
similarity index 100%
rename from src/classifiers/KDBLd.cc
rename to bayesnet/classifiers/KDBLd.cc
diff --git a/src/classifiers/KDBLd.h b/bayesnet/classifiers/KDBLd.h
similarity index 100%
rename from src/classifiers/KDBLd.h
rename to bayesnet/classifiers/KDBLd.h
index bc98a3e..3597576 100644
--- a/src/classifiers/KDBLd.h
+++ b/bayesnet/classifiers/KDBLd.h
@@ -1,7 +1,7 @@
 #ifndef KDBLD_H
 #define KDBLD_H
-#include "KDB.h"
 #include "Proposal.h"
+#include "KDB.h"
 
 namespace bayesnet {
     class KDBLd : public KDB, public Proposal {
diff --git a/src/classifiers/Proposal.cc b/bayesnet/classifiers/Proposal.cc
similarity index 100%
rename from src/classifiers/Proposal.cc
rename to bayesnet/classifiers/Proposal.cc
index 5687eba..aef27bf 100644
--- a/src/classifiers/Proposal.cc
+++ b/bayesnet/classifiers/Proposal.cc
@@ -1,5 +1,5 @@
-#include "Proposal.h"
 #include <ArffFiles.h>
+#include "Proposal.h"
 
 namespace bayesnet {
     Proposal::Proposal(torch::Tensor& dataset_, std::vector<std::string>& features_, std::string& className_) : pDataset(dataset_), pFeatures(features_), pClassName(className_) {}
diff --git a/src/classifiers/Proposal.h b/bayesnet/classifiers/Proposal.h
similarity index 96%
rename from src/classifiers/Proposal.h
rename to bayesnet/classifiers/Proposal.h
index 37d8a26..59e8451 100644
--- a/src/classifiers/Proposal.h
+++ b/bayesnet/classifiers/Proposal.h
@@ -3,8 +3,8 @@
 #include <string>
 #include <map>
 #include <torch/torch.h>
-#include "bayesian_network/Network.h"
 #include <CPPFImdlp.h>
+#include "bayesnet/network/Network.h"
 #include "Classifier.h"
 
 namespace bayesnet {
diff --git a/src/classifiers/SPODE.cc b/bayesnet/classifiers/SPODE.cc
similarity index 100%
rename from src/classifiers/SPODE.cc
rename to bayesnet/classifiers/SPODE.cc
diff --git a/src/classifiers/SPODE.h b/bayesnet/classifiers/SPODE.h
similarity index 100%
rename from src/classifiers/SPODE.h
rename to bayesnet/classifiers/SPODE.h
diff --git a/src/classifiers/SPODELd.cc b/bayesnet/classifiers/SPODELd.cc
similarity index 100%
rename from src/classifiers/SPODELd.cc
rename to bayesnet/classifiers/SPODELd.cc
diff --git a/src/classifiers/SPODELd.h b/bayesnet/classifiers/SPODELd.h
similarity index 100%
rename from src/classifiers/SPODELd.h
rename to bayesnet/classifiers/SPODELd.h
diff --git a/src/classifiers/TAN.cc b/bayesnet/classifiers/TAN.cc
similarity index 100%
rename from src/classifiers/TAN.cc
rename to bayesnet/classifiers/TAN.cc
diff --git a/src/classifiers/TAN.h b/bayesnet/classifiers/TAN.h
similarity index 100%
rename from src/classifiers/TAN.h
rename to bayesnet/classifiers/TAN.h
diff --git a/src/classifiers/TANLd.cc b/bayesnet/classifiers/TANLd.cc
similarity index 100%
rename from src/classifiers/TANLd.cc
rename to bayesnet/classifiers/TANLd.cc
diff --git a/src/classifiers/TANLd.h b/bayesnet/classifiers/TANLd.h
similarity index 100%
rename from src/classifiers/TANLd.h
rename to bayesnet/classifiers/TANLd.h
diff --git a/src/ensembles/AODE.cc b/bayesnet/ensembles/AODE.cc
similarity index 100%
rename from src/ensembles/AODE.cc
rename to bayesnet/ensembles/AODE.cc
diff --git a/src/ensembles/AODE.h b/bayesnet/ensembles/AODE.h
similarity index 91%
rename from src/ensembles/AODE.h
rename to bayesnet/ensembles/AODE.h
index 592b1e8..bceffd5 100644
--- a/src/ensembles/AODE.h
+++ b/bayesnet/ensembles/AODE.h
@@ -1,7 +1,7 @@
 #ifndef AODE_H
 #define AODE_H
+#include "bayesnet/classifiers/SPODE.h"
 #include "Ensemble.h"
-#include "classifiers/SPODE.h"
 namespace bayesnet {
     class AODE : public Ensemble {
     public:
diff --git a/src/ensembles/AODELd.cc b/bayesnet/ensembles/AODELd.cc
similarity index 100%
rename from src/ensembles/AODELd.cc
rename to bayesnet/ensembles/AODELd.cc
diff --git a/src/ensembles/AODELd.h b/bayesnet/ensembles/AODELd.h
similarity index 89%
rename from src/ensembles/AODELd.h
rename to bayesnet/ensembles/AODELd.h
index b8d2fe3..178161c 100644
--- a/src/ensembles/AODELd.h
+++ b/bayesnet/ensembles/AODELd.h
@@ -1,8 +1,8 @@
 #ifndef AODELD_H
 #define AODELD_H
+#include "bayesnet/classifiers/Proposal.h"
+#include "bayesnet/classifiers/SPODELd.h"
 #include "Ensemble.h"
-#include "classifiers/Proposal.h"
-#include "classifiers/SPODELd.h"
 
 namespace bayesnet {
     class AODELd : public Ensemble, public Proposal {
diff --git a/src/ensembles/BoostAODE.cc b/bayesnet/ensembles/BoostAODE.cc
similarity index 99%
rename from src/ensembles/BoostAODE.cc
rename to bayesnet/ensembles/BoostAODE.cc
index 202b02c..1d16f1c 100644
--- a/src/ensembles/BoostAODE.cc
+++ b/bayesnet/ensembles/BoostAODE.cc
@@ -3,10 +3,10 @@
 #include <limits.h>
 #include <tuple>
 #include <folding.hpp>
+#include "bayesnet/feature_selection/CFS.h"
+#include "bayesnet/feature_selection/FCBF.h"
+#include "bayesnet/feature_selection/IWSS.h"
 #include "BoostAODE.h"
-#include "feature_selection/CFS.h"
-#include "feature_selection/FCBF.h"
-#include "feature_selection/IWSS.h"
 
 namespace bayesnet {
     struct {
diff --git a/src/ensembles/BoostAODE.h b/bayesnet/ensembles/BoostAODE.h
similarity index 94%
rename from src/ensembles/BoostAODE.h
rename to bayesnet/ensembles/BoostAODE.h
index e21b38a..2c0ed14 100644
--- a/src/ensembles/BoostAODE.h
+++ b/bayesnet/ensembles/BoostAODE.h
@@ -1,9 +1,9 @@
 #ifndef BOOSTAODE_H
 #define BOOSTAODE_H
-#include "Ensemble.h"
 #include <map>
-#include "classifiers/SPODE.h"
-#include "feature_selection/FeatureSelect.h"
+#include "bayesnet/classifiers/SPODE.h"
+#include "bayesnet/feature_selection/FeatureSelect.h"
+#include "Ensemble.h"
 namespace bayesnet {
     class BoostAODE : public Ensemble {
     public:
diff --git a/src/ensembles/Ensemble.cc b/bayesnet/ensembles/Ensemble.cc
similarity index 100%
rename from src/ensembles/Ensemble.cc
rename to bayesnet/ensembles/Ensemble.cc
diff --git a/src/ensembles/Ensemble.h b/bayesnet/ensembles/Ensemble.h
similarity index 93%
rename from src/ensembles/Ensemble.h
rename to bayesnet/ensembles/Ensemble.h
index cf021ae..cb4220a 100644
--- a/src/ensembles/Ensemble.h
+++ b/bayesnet/ensembles/Ensemble.h
@@ -1,9 +1,9 @@
 #ifndef ENSEMBLE_H
 #define ENSEMBLE_H
 #include <torch/torch.h>
-#include "classifiers/Classifier.h"
-#include "utils/BayesMetrics.h"
-#include "utils/bayesnetUtils.h"
+#include "bayesnet/utils/BayesMetrics.h"
+#include "bayesnet/utils/bayesnetUtils.h"
+#include "bayesnet/classifiers/Classifier.h"
 
 namespace bayesnet {
     class Ensemble : public Classifier {
diff --git a/src/feature_selection/CFS.cc b/bayesnet/feature_selection/CFS.cc
similarity index 98%
rename from src/feature_selection/CFS.cc
rename to bayesnet/feature_selection/CFS.cc
index b17b4f2..bf0c746 100644
--- a/src/feature_selection/CFS.cc
+++ b/bayesnet/feature_selection/CFS.cc
@@ -1,6 +1,6 @@
-#include "CFS.h"
 #include <limits>
-#include "utils/bayesnetUtils.h"
+#include "bayesnet/utils/bayesnetUtils.h"
+#include "CFS.h"
 namespace bayesnet {
     void CFS::fit()
     {
diff --git a/src/feature_selection/CFS.h b/bayesnet/feature_selection/CFS.h
similarity index 92%
rename from src/feature_selection/CFS.h
rename to bayesnet/feature_selection/CFS.h
index 5b718f3..0d44e2b 100644
--- a/src/feature_selection/CFS.h
+++ b/bayesnet/feature_selection/CFS.h
@@ -2,7 +2,7 @@
 #define CFS_H
 #include <torch/torch.h>
 #include <vector>
-#include "feature_selection/FeatureSelect.h"
+#include "bayesnet/feature_selection/FeatureSelect.h"
 namespace bayesnet {
     class CFS : public FeatureSelect {
     public:
diff --git a/src/feature_selection/FCBF.cc b/bayesnet/feature_selection/FCBF.cc
similarity index 97%
rename from src/feature_selection/FCBF.cc
rename to bayesnet/feature_selection/FCBF.cc
index 1e37722..ebf6d7a 100644
--- a/src/feature_selection/FCBF.cc
+++ b/bayesnet/feature_selection/FCBF.cc
@@ -1,4 +1,4 @@
-#include "utils/bayesnetUtils.h"
+#include "bayesnet/utils/bayesnetUtils.h"
 #include "FCBF.h"
 namespace bayesnet {
 
diff --git a/src/feature_selection/FCBF.h b/bayesnet/feature_selection/FCBF.h
similarity index 91%
rename from src/feature_selection/FCBF.h
rename to bayesnet/feature_selection/FCBF.h
index c323664..8ec3a8f 100644
--- a/src/feature_selection/FCBF.h
+++ b/bayesnet/feature_selection/FCBF.h
@@ -2,7 +2,7 @@
 #define FCBF_H
 #include <torch/torch.h>
 #include <vector>
-#include "feature_selection/FeatureSelect.h"
+#include "bayesnet/feature_selection/FeatureSelect.h"
 namespace bayesnet {
     class FCBF : public FeatureSelect {
     public:
diff --git a/src/feature_selection/FeatureSelect.cc b/bayesnet/feature_selection/FeatureSelect.cc
similarity index 98%
rename from src/feature_selection/FeatureSelect.cc
rename to bayesnet/feature_selection/FeatureSelect.cc
index 29c0643..54dcfb0 100644
--- a/src/feature_selection/FeatureSelect.cc
+++ b/bayesnet/feature_selection/FeatureSelect.cc
@@ -1,6 +1,6 @@
-#include "FeatureSelect.h"
 #include <limits>
-#include "utils/bayesnetUtils.h"
+#include "bayesnet/utils/bayesnetUtils.h"
+#include "FeatureSelect.h"
 namespace bayesnet {
     FeatureSelect::FeatureSelect(const torch::Tensor& samples, const std::vector<std::string>& features, const std::string& className, const int maxFeatures, const int classNumStates, const torch::Tensor& weights) :
         Metrics(samples, features, className, classNumStates), maxFeatures(maxFeatures == 0 ? samples.size(0) - 1 : maxFeatures), weights(weights)
diff --git a/src/feature_selection/FeatureSelect.h b/bayesnet/feature_selection/FeatureSelect.h
similarity index 96%
rename from src/feature_selection/FeatureSelect.h
rename to bayesnet/feature_selection/FeatureSelect.h
index a697dd7..fbd3abc 100644
--- a/src/feature_selection/FeatureSelect.h
+++ b/bayesnet/feature_selection/FeatureSelect.h
@@ -2,7 +2,7 @@
 #define FEATURE_SELECT_H
 #include <torch/torch.h>
 #include <vector>
-#include "utils/BayesMetrics.h"
+#include "bayesnet/utils/BayesMetrics.h"
 namespace bayesnet {
     class FeatureSelect : public Metrics {
     public:
diff --git a/src/feature_selection/IWSS.cc b/bayesnet/feature_selection/IWSS.cc
similarity index 97%
rename from src/feature_selection/IWSS.cc
rename to bayesnet/feature_selection/IWSS.cc
index 80cd89e..8ed3d76 100644
--- a/src/feature_selection/IWSS.cc
+++ b/bayesnet/feature_selection/IWSS.cc
@@ -1,6 +1,6 @@
-#include "IWSS.h"
 #include <limits>
-#include "utils/bayesnetUtils.h"
+#include "bayesnet/utils/bayesnetUtils.h"
+#include "IWSS.h"
 namespace bayesnet {
     IWSS::IWSS(const torch::Tensor& samples, const std::vector<std::string>& features, const std::string& className, const int maxFeatures, const int classNumStates, const torch::Tensor& weights, const double threshold) :
         FeatureSelect(samples, features, className, maxFeatures, classNumStates, weights), threshold(threshold)
diff --git a/src/feature_selection/IWSS.h b/bayesnet/feature_selection/IWSS.h
similarity index 100%
rename from src/feature_selection/IWSS.h
rename to bayesnet/feature_selection/IWSS.h
index 0faea7d..a990e60 100644
--- a/src/feature_selection/IWSS.h
+++ b/bayesnet/feature_selection/IWSS.h
@@ -1,7 +1,7 @@
 #ifndef IWSS_H
 #define IWSS_H
-#include <torch/torch.h>
 #include <vector>
+#include <torch/torch.h>
 #include "FeatureSelect.h"
 namespace bayesnet {
     class IWSS : public FeatureSelect {
diff --git a/src/bayesian_network/Network.cc b/bayesnet/network/Network.cc
similarity index 99%
rename from src/bayesian_network/Network.cc
rename to bayesnet/network/Network.cc
index 6e7e8f1..e148fa0 100644
--- a/src/bayesian_network/Network.cc
+++ b/bayesnet/network/Network.cc
@@ -1,7 +1,7 @@
 #include <thread>
 #include <mutex>
 #include "Network.h"
-#include "utils/bayesnetUtils.h"
+#include "bayesnet/utils/bayesnetUtils.h"
 namespace bayesnet {
     Network::Network() : features(std::vector<std::string>()), className(""), classNumStates(0), fitted(false), laplaceSmoothing(0) {}
     Network::Network(float maxT) : features(std::vector<std::string>()), className(""), classNumStates(0), maxThreads(maxT), fitted(false), laplaceSmoothing(0) {}
diff --git a/src/bayesian_network/Network.h b/bayesnet/network/Network.h
similarity index 99%
rename from src/bayesian_network/Network.h
rename to bayesnet/network/Network.h
index ad45055..6994fe0 100644
--- a/src/bayesian_network/Network.h
+++ b/bayesnet/network/Network.h
@@ -1,9 +1,9 @@
 #ifndef NETWORK_H
 #define NETWORK_H
-#include "Node.h"
 #include <map>
 #include <vector>
-#include "config.h"
+#include "bayesnet/config.h"
+#include "Node.h"
 
 namespace bayesnet {
     class Network {
diff --git a/src/bayesian_network/Node.cc b/bayesnet/network/Node.cc
similarity index 100%
rename from src/bayesian_network/Node.cc
rename to bayesnet/network/Node.cc
diff --git a/src/bayesian_network/Node.h b/bayesnet/network/Node.h
similarity index 100%
rename from src/bayesian_network/Node.h
rename to bayesnet/network/Node.h
index a49158f..81f6f73 100644
--- a/src/bayesian_network/Node.h
+++ b/bayesnet/network/Node.h
@@ -1,9 +1,9 @@
 #ifndef NODE_H
 #define NODE_H
-#include <torch/torch.h>
 #include <unordered_set>
 #include <vector>
 #include <string>
+#include <torch/torch.h>
 namespace bayesnet {
     class Node {
     private:
diff --git a/src/utils/BayesMetrics.cc b/bayesnet/utils/BayesMetrics.cc
similarity index 100%
rename from src/utils/BayesMetrics.cc
rename to bayesnet/utils/BayesMetrics.cc
index 4e2633f..4617581 100644
--- a/src/utils/BayesMetrics.cc
+++ b/bayesnet/utils/BayesMetrics.cc
@@ -1,5 +1,5 @@
-#include "BayesMetrics.h"
 #include "Mst.h"
+#include "BayesMetrics.h"
 namespace bayesnet {
     //samples is n+1xm tensor used to fit the model
     Metrics::Metrics(const torch::Tensor& samples, const std::vector<std::string>& features, const std::string& className, const int classNumStates)
diff --git a/src/utils/BayesMetrics.h b/bayesnet/utils/BayesMetrics.h
similarity index 100%
rename from src/utils/BayesMetrics.h
rename to bayesnet/utils/BayesMetrics.h
index 5d2b927..12a22c7 100644
--- a/src/utils/BayesMetrics.h
+++ b/bayesnet/utils/BayesMetrics.h
@@ -1,8 +1,8 @@
 #ifndef BAYESNET_METRICS_H
 #define BAYESNET_METRICS_H
-#include <torch/torch.h>
 #include <vector>
 #include <string>
+#include <torch/torch.h>
 namespace bayesnet {
     class Metrics {
     private:
diff --git a/src/utils/Mst.cc b/bayesnet/utils/Mst.cc
similarity index 100%
rename from src/utils/Mst.cc
rename to bayesnet/utils/Mst.cc
index 41207e8..b0897b2 100644
--- a/src/utils/Mst.cc
+++ b/bayesnet/utils/Mst.cc
@@ -1,6 +1,6 @@
-#include "Mst.h"
 #include <vector>
 #include <list>
+#include "Mst.h"
 /*
     Based on the code from https://www.softwaretestinghelp.com/minimum-spanning-tree-tutorial/
 
diff --git a/src/utils/Mst.h b/bayesnet/utils/Mst.h
similarity index 100%
rename from src/utils/Mst.h
rename to bayesnet/utils/Mst.h
index 8e7c462..9fa951b 100644
--- a/src/utils/Mst.h
+++ b/bayesnet/utils/Mst.h
@@ -1,8 +1,8 @@
 #ifndef MST_H
 #define MST_H
-#include <torch/torch.h>
 #include <vector>
 #include <string>
+#include <torch/torch.h>
 namespace bayesnet {
     class MST {
     private:
diff --git a/src/utils/bayesnetUtils.cc b/bayesnet/utils/bayesnetUtils.cc
similarity index 100%
rename from src/utils/bayesnetUtils.cc
rename to bayesnet/utils/bayesnetUtils.cc
diff --git a/src/utils/bayesnetUtils.h b/bayesnet/utils/bayesnetUtils.h
similarity index 100%
rename from src/utils/bayesnetUtils.h
rename to bayesnet/utils/bayesnetUtils.h
index 0e741be..8b18974 100644
--- a/src/utils/bayesnetUtils.h
+++ b/bayesnet/utils/bayesnetUtils.h
@@ -1,7 +1,7 @@
 #ifndef BAYESNET_UTILS_H
 #define BAYESNET_UTILS_H
-#include <torch/torch.h>
 #include <vector>
+#include <torch/torch.h>
 namespace bayesnet {
     std::vector<int> argsort(std::vector<double>& nums);
     std::vector<std::vector<int>> tensorToVector(torch::Tensor& dtensor);
diff --git a/gcovr.cfg b/gcovr.cfg
index 01ad57e..1739d47 100644
--- a/gcovr.cfg
+++ b/gcovr.cfg
@@ -1,4 +1,4 @@
-filter = src/
+filter = bayesnet/
 exclude-directories = build_debug/lib/
 print-summary = yes
 sort = uncovered-percent
diff --git a/sample/CMakeLists.txt b/sample/CMakeLists.txt
index b5f90e0..85486ad 100644
--- a/sample/CMakeLists.txt
+++ b/sample/CMakeLists.txt
@@ -11,7 +11,7 @@ include_directories(
     lib/Files
     lib/mdlp
     lib/json/include
-    /usr/local/include/bayesnet
+    /usr/local/include/
 )
 
 add_subdirectory(lib/Files)
diff --git a/sample/sample.cc b/sample/sample.cc
index 848f4a6..a6352e5 100644
--- a/sample/sample.cc
+++ b/sample/sample.cc
@@ -1,6 +1,6 @@
 #include <ArffFiles.h>
 #include <CPPFImdlp.h>
-#include <ensembles/BoostAODE.h>
+#include <bayesnet/ensembles/BoostAODE.h>
 
 std::vector<mdlp::labels_t> discretizeDataset(std::vector<mdlp::samples_t>& X, mdlp::labels_t& y)
 {
diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt
index 630beab..7613253 100644
--- a/tests/CMakeLists.txt
+++ b/tests/CMakeLists.txt
@@ -1,19 +1,14 @@
 if(ENABLE_TESTING)
     set(TEST_BAYESNET "unit_tests_bayesnet")
     include_directories(
-        ${BayesNet_SOURCE_DIR}/src
-        ${BayesNet_SOURCE_DIR}/src/feature_selection
-        ${BayesNet_SOURCE_DIR}/src/bayesian_network
-        ${BayesNet_SOURCE_DIR}/src/classifiers
-        ${BayesNet_SOURCE_DIR}/src/utils
-        ${BayesNet_SOURCE_DIR}/src/ensembles
         ${BayesNet_SOURCE_DIR}/lib/Files
         ${BayesNet_SOURCE_DIR}/lib/mdlp
         ${BayesNet_SOURCE_DIR}/lib/folding
         ${BayesNet_SOURCE_DIR}/lib/json/include
+        ${BayesNet_SOURCE_DIR}
         ${CMAKE_BINARY_DIR}/configured_files/include
     )
-    file(GLOB_RECURSE BayesNet_SOURCES "${BayesNet_SOURCE_DIR}/src/*.cc")
+    file(GLOB_RECURSE BayesNet_SOURCES "${BayesNet_SOURCE_DIR}/bayesnet/*.cc")
     set(TEST_SOURCES_BAYESNET TestBayesModels.cc TestBayesNetwork.cc TestBayesMetrics.cc TestUtils.cc ${BayesNet_SOURCES})
     add_executable(${TEST_BAYESNET} ${TEST_SOURCES_BAYESNET})
     target_link_libraries(${TEST_BAYESNET} PUBLIC "${TORCH_LIBRARIES}" ArffFiles mdlp Catch2::Catch2WithMain )
diff --git a/tests/TestBayesMetrics.cc b/tests/TestBayesMetrics.cc
index 5c34173..1b1b815 100644
--- a/tests/TestBayesMetrics.cc
+++ b/tests/TestBayesMetrics.cc
@@ -1,7 +1,7 @@
 #include <catch2/catch_test_macros.hpp>
 #include <catch2/catch_approx.hpp>
 #include <catch2/generators/catch_generators.hpp>
-#include "BayesMetrics.h"
+#include "bayesnet/utils/BayesMetrics.h"
 #include "TestUtils.h"
 
 
diff --git a/tests/TestBayesModels.cc b/tests/TestBayesModels.cc
index 5d8d3d5..58a773b 100644
--- a/tests/TestBayesModels.cc
+++ b/tests/TestBayesModels.cc
@@ -2,15 +2,15 @@
 #include <catch2/catch_test_macros.hpp>
 #include <catch2/catch_approx.hpp>
 #include <catch2/generators/catch_generators.hpp>
-#include "KDB.h"
-#include "TAN.h"
-#include "SPODE.h"
-#include "AODE.h"
-#include "BoostAODE.h"
-#include "TANLd.h"
-#include "KDBLd.h"
-#include "SPODELd.h"
-#include "AODELd.h"
+#include "bayesnet/classifiers/KDB.h"
+#include "bayesnet/classifiers/TAN.h"
+#include "bayesnet/classifiers/SPODE.h"
+#include "bayesnet/classifiers/TANLd.h"
+#include "bayesnet/classifiers/KDBLd.h"
+#include "bayesnet/classifiers/SPODELd.h"
+#include "bayesnet/ensembles/AODE.h"
+#include "bayesnet/ensembles/AODELd.h"
+#include "bayesnet/ensembles/BoostAODE.h"
 #include "TestUtils.h"
 
 const std::string ACTUAL_VERSION = "1.0.4";
diff --git a/tests/TestBayesNetwork.cc b/tests/TestBayesNetwork.cc
index 795ed13..e1ef531 100644
--- a/tests/TestBayesNetwork.cc
+++ b/tests/TestBayesNetwork.cc
@@ -3,7 +3,7 @@
 #include <catch2/generators/catch_generators.hpp>
 #include <string>
 #include "TestUtils.h"
-#include "Network.h"
+#include "bayesnet/network/Network.h"
 
 void buildModel(bayesnet::Network& net, const std::vector<std::string>& features, const std::string& className)
 {
diff --git a/tests/TestUtils.cc b/tests/TestUtils.cc
index 5a0910e..64f5769 100644
--- a/tests/TestUtils.cc
+++ b/tests/TestUtils.cc
@@ -1,5 +1,5 @@
 #include "TestUtils.h"
-#include "config.h"
+#include "bayesnet/config.h"
 
 class Paths {
 public: