2 files changed, 4 insertions, 3 deletions
diff --git a/include/brain/neural_network.h b/include/brain/neural_network.h
index b2c441f..ccd65eb 100644
--- a/include/brain/neural_network.h
+++ b/include/brain/neural_network.h
@@ -14,7 +14,8 @@ class NeuralNetwork
 {
 public:
     NeuralNetwork(std::istream &&networkConfigFile, int numberOfSensors, int numberOfOutputs);
-
+    NeuralNetwork(std::istream &&networkConfigFile, std::vector<bool> sensorInitialValues, int numberOfOutputs);
+    
     void setInput(int inputIndex, double activation);
     int findMaxOutputIndex() const;
 
diff --git a/include/spacebot.h b/include/spacebot.h
index 079b33e..9f7f83e 100644
--- a/include/spacebot.h
+++ b/include/spacebot.h
@@ -6,11 +6,11 @@
 
 class Spacebot {
 public:
-    Spacebot(std::string outputPath);
+    Spacebot(std::string outputPath, std::string brainFilename);
     void writeNextMove();
 private:
     std::string _outputFilename;
-    std::string _networkConfigFilename;
+    std::string _brainFilename;
     GameState _gameState;
     
     void writeMove(const Move& move);