diff --git a/src/machineLearning/machineLearning.cpp b/src/machineLearning/machineLearning.cpp
index 19f141fef1500a5e639482d6ea3d531154d28d04..7f5219d1faf2be063367620447e582229376de22 100644
--- a/src/machineLearning/machineLearning.cpp
+++ b/src/machineLearning/machineLearning.cpp
@@ -10,6 +10,15 @@
 
 namespace rapidmix {
     
+    //////////////////////////////////////////////////////////////////////// Generic train
+    template <class MachineLearningModule>
+    bool machineLearning<MachineLearningModule>::train(const trainingData &newTrainingData) {
+        return MachineLearningModule::train(newTrainingData);
+    }
+
+    
+    /////////////////////////////////////////////////////////////////////// RapidLib specializations
+    
     void trainingData2rapidLib (const trainingData &newTrainingData, std::vector<trainingExample<double> > &trainingSet) {
         for (int h = 0; h < newTrainingData.trainingSet.size(); ++h) { //Go through every phrase
             
@@ -26,6 +35,7 @@ namespace rapidmix {
         }
     };
     
+    /////////////////////////////////////////////////////////////////////// RapidLib classification
     template<>
     bool machineLearning<classification<double>>::train(const trainingData &newTrainingData) {
         std::vector<trainingExample<double> > trainingSet;
@@ -37,6 +47,7 @@ namespace rapidmix {
         return classification::train(trainingSet);
     }
     
+    /////////////////////////////////////////////////////////////////////// RapidLib regression
     template<>
     bool machineLearning<regression<double> >::train(const trainingData &newTrainingData) {
         std::vector<trainingExample<double> > trainingSet;
@@ -44,6 +55,7 @@ namespace rapidmix {
         return regression::train(trainingSet);
     }
     
+    /////////////////////////////////////////////////////////////////////// RapidLib seriesClassification
     template<>
     bool machineLearning<seriesClassification<double> >::train(const trainingData &newTrainingData) {
         std::vector<trainingSeries<double> > seriesSet;
@@ -58,18 +70,6 @@ namespace rapidmix {
         return seriesClassification::train(seriesSet);
     }
     
-    template<>
-    bool machineLearning<rapidGVF>::train(const trainingData &newTrainingData) {
-        return rapidGVF::train(newTrainingData);
-    }
-    
-    
-    ///Generic train
-    template <class MachineLearningModule>
-    bool machineLearning<MachineLearningModule>::train(const trainingData &newTrainingData) {
-        return MachineLearningModule::train(newTrainingData);
-    }
-    
     template<>
     std::string machineLearning<classification<double> >::run(const std::vector<double> &inputVector, const std::string &label) {
         int classIndex = classification::run(inputVector)[0];
@@ -81,4 +81,11 @@ namespace rapidmix {
         return seriesClassification::run(inputSeries);
     }
     
+    
+    /////////////////////////////////////////////////////////////////////// GVF
+    template<>
+    bool machineLearning<rapidGVF>::train(const trainingData &newTrainingData) {
+        return rapidGVF::train(newTrainingData);
+    }
+    
 }
\ No newline at end of file
diff --git a/src/machineLearning/machineLearning.h b/src/machineLearning/machineLearning.h
index 97d6fd3515fb74b65284252e439eded0e7ca0c90..85875e174249102a454ee95cbe15edc58caaaf0e 100644
--- a/src/machineLearning/machineLearning.h
+++ b/src/machineLearning/machineLearning.h
@@ -10,7 +10,7 @@
 #ifndef machineLearning_h
 #define machineLearning_h
 
-#include "rapidMix.h"
+#include "../rapidMix.h"
 
 ////////// Include all of the machine learning algorithms here
 #include "classification.h"
diff --git a/src/machineLearning/trainingData.h b/src/machineLearning/trainingData.h
index e9c12eb26fd0a6b21b9383d11990df04c6c8df9e..e1dcd9c4ee5c6d588c85acbca0bea526d9d58b12 100644
--- a/src/machineLearning/trainingData.h
+++ b/src/machineLearning/trainingData.h
@@ -15,7 +15,7 @@
 #include <string>
 #include <unordered_map>
 
-#include "rapidMix.h"
+#include "../rapidMix.h"
 #include "json.h"
 
 namespace rapidmix {