diff --git a/src/Computation.cxx b/src/Computation.cxx
index 8f7b89e..f928040 100644
--- a/src/Computation.cxx
+++ b/src/Computation.cxx
@@ -634,6 +634,14 @@ void Computation::setPropagationModel(PropagationModel *propagationModel){
     this->propagationModel = propagationModel;
 }
 
+
+void Computation::setInputAugmented(const std::vector<double>& newInputAugmented){
+    if(newInputAugmented.size() == inputAugmented.size()){
+        inputAugmented = newInputAugmented;
+        InputAugmentedArma = arma::Col<double>(newInputAugmented);
+    }
+}
+
 void Computation::resetVirtualOutputs(){
     //virtual outputs have the names starting with v-out:
     for(auto it = nodeToIndex.cbegin(); it!=nodeToIndex.cend();it++){
diff --git a/src/Computation.h b/src/Computation.h
index 78c84f4..356424d 100644
--- a/src/Computation.h
+++ b/src/Computation.h
@@ -130,6 +130,7 @@ class Computation{
         void setDissipationModel(DissipationModel* dissipationModel);
         void setConservationModel(ConservationModel* conservationModel);
         void setPropagationModel(PropagationModel* propagationModel);
+        void setInputAugmented(const std::vector<double>& inputAugmented);
 
         // get-set for saturation function
         std::function<double(double,double)> getSaturationFunction()const{return saturationFunction;}