diff --git a/nicolas-train/src/main/java/pl/waw/ipipan/zil/summ/nicolas/train/Main.java b/nicolas-train/src/main/java/pl/waw/ipipan/zil/summ/nicolas/train/Prepare.java
index 3c021fc..f42acab 100644
--- a/nicolas-train/src/main/java/pl/waw/ipipan/zil/summ/nicolas/train/Main.java
+++ b/nicolas-train/src/main/java/pl/waw/ipipan/zil/summ/nicolas/train/Prepare.java
@@ -2,9 +2,9 @@ package pl.waw.ipipan.zil.summ.nicolas.train;
 
 import pl.waw.ipipan.zil.summ.nicolas.train.pipeline.*;
 
-public class Main {
+public class Prepare {
 
-    private Main() {
+    private Prepare() {
     }
 
     public static void main(String[] args) throws Exception {
@@ -13,7 +13,5 @@ public class Main {
         ExtractGoldSummaries.main(args);
         CreateOptimalSummaries.main(args);
         ExtractMostFrequentMentions.main(args);
-        PrepareTrainingData.main(args);
-        TrainAllModels.main(args);
     }
 }
diff --git a/nicolas-train/src/main/java/pl/waw/ipipan/zil/summ/nicolas/train/Train.java b/nicolas-train/src/main/java/pl/waw/ipipan/zil/summ/nicolas/train/Train.java
new file mode 100644
index 0000000..6acc765
--- /dev/null
+++ b/nicolas-train/src/main/java/pl/waw/ipipan/zil/summ/nicolas/train/Train.java
@@ -0,0 +1,15 @@
+package pl.waw.ipipan.zil.summ.nicolas.train;
+
+import pl.waw.ipipan.zil.summ.nicolas.train.pipeline.PrepareTrainingData;
+import pl.waw.ipipan.zil.summ.nicolas.train.pipeline.TrainAllModels;
+
+public class Train {
+
+    private Train() {
+    }
+
+    public static void main(String[] args) throws Exception {
+        PrepareTrainingData.main(args);
+        TrainAllModels.main(args);
+    }
+}
diff --git a/train.sh b/train.sh
index 46b5624..256fa8e 100755
--- a/train.sh
+++ b/train.sh
@@ -1,5 +1,7 @@
 #!/usr/bin/env bash
 
 mvn clean install -Dmaven.test.skip=true
-mvn -pl nicolas-train exec:java -Dexec.mainClass="pl.waw.ipipan.zil.summ.nicolas.train.Main"
+mvn -pl nicolas-train exec:java -Dexec.mainClass="pl.waw.ipipan.zil.summ.nicolas.train.Prepare"
+mvn install -Dmaven.test.skip=true
+mvn -pl nicolas-train exec:java -Dexec.mainClass="pl.waw.ipipan.zil.summ.nicolas.train.Train"
 mvn install -Dmaven.test.skip=true