diff --git a/source/main.cpp b/source/main.cpp
index 47e04a3..987c65c 100644
--- a/source/main.cpp
+++ b/source/main.cpp
@@ -105,6 +105,9 @@ int main()
 	stack.save();
 
 	// Shake weights
+	stack.weightsInit(0.01);
+
+	// Save weights
 	stack.saveWeights();
 
 #else	
@@ -116,7 +119,6 @@ int main()
 #endif	
 	
 	// Train stack
-	stack.weightsInit(0.01);
 	stack.train(batch, 100, 1000, &statusDisplay);
 
 	// Save weights