diff --git a/sample/CMakeLists.txt b/sample/CMakeLists.txt
index f8ccf8d..f062ca4 100644
--- a/sample/CMakeLists.txt
+++ b/sample/CMakeLists.txt
@@ -1,5 +1,5 @@
 include_directories(
-    ${Platform_SOURCE_DIR}/src/Platform
+    ${Platform_SOURCE_DIR}/src/Platform/modules
     ${Platform_SOURCE_DIR}/lib/PyClassifiers/src/PyClassifiers
     ${Python3_INCLUDE_DIRS}
     ${Platform_SOURCE_DIR}/lib/Files
diff --git a/sample/sample.cc b/sample/sample.cc
index b96436f..b3f95a6 100644
--- a/sample/sample.cc
+++ b/sample/sample.cc
@@ -195,6 +195,7 @@ int main(int argc, char** argv)
     }
     float total_score = 0, total_score_train = 0, score_train, score_test;
     folding::Fold* fold;
+    double nodes = 0.0;
     if (stratified)
         fold = new folding::StratifiedKFold(nFolds, y, seed);
     else
@@ -217,6 +218,8 @@ int main(int argc, char** argv)
             auto [Xtrain, ytrain] = extract_indices(train, Xd, y);
             auto [Xtest, ytest] = extract_indices(test, Xd, y);
             clf->fit(Xtrain, ytrain, features, className, states);
+            std::cout << "Nodes: " << clf->getNumberOfNodes() << std::endl;
+            nodes += clf->getNumberOfNodes();
             score_train = clf->score(Xtrain, ytrain);
             score_test = clf->score(Xtest, ytest);
         }
@@ -230,6 +233,7 @@ int main(int argc, char** argv)
         std::cout << "Score Test : " << score_test << std::endl;
         std::cout << "-------------------------------------------------------------------------------" << std::endl;
     }
+    std::cout << "Nodes: " << nodes / nFolds << std::endl;
     std::cout << "**********************************************************************************" << std::endl;
     std::cout << "Average Score Train: " << total_score_train / nFolds << std::endl;
     std::cout << "Average Score Test : " << total_score / nFolds << std::endl;return 0;