1 files changed, 9 insertions, 2 deletions
diff --git a/main/main.cpp b/main/main.cpp
index 578620c..ce6bc43 100644
--- a/main/main.cpp
+++ b/main/main.cpp
@@ -5,12 +5,19 @@
 
 int main(int argc, char* argv[])
 {
-    if (argc < 2) {
+    if (argc < 2)
+    {
 	std::cout << "usage: " << argv[0] << " <output folder>" << std::endl;
 	return 1;
     }
 
-    Spacebot bot(argv[1]);
+    std::string brainFilename = "brain.nn";
+    if (argc >= 3)
+    {
+	brainFilename = argv[2];
+    }
+
+    Spacebot bot(argv[1], std::move(brainFilename));
     bot.writeNextMove();
     
     return 0;