diff --git a/demo/training_configuration.json b/demo/training_configuration.json
index 367daa6fb..c31d7b186 100644
--- a/demo/training_configuration.json
+++ b/demo/training_configuration.json
@@ -6,11 +6,11 @@
      },
      "network configuration": {
          "skip connections" : false,
-         "nodes per layer" : [7,32,32,32,5],
+         "nodes per layer" : [11,32,32,32,9],
          "activation function" : "sigmoid"
      },
      "tensor names": {
-         "inputs"  : ["potential_temperature","qv", "qc", "qr", "qs", "pressure", "temperature"],
-         "outputs" : ["potential_temperature","qv", "qc", "qr", "qs"]
+         "inputs"  : ["potential_temperature","qv", "qc", "qr", "qs", "pressure", "temperature", "ni", "nr", "qi", "qg"],
+         "outputs" : ["potential_temperature","qv", "qc", "qr", "qs", "ni", "nr", "qi", "qg"]
      }
  }
diff --git a/demo/training_data_files.json b/demo/training_data_files.json
index d88ffee0b..39318c9bb 100644
--- a/demo/training_data_files.json
+++ b/demo/training_data_files.json
@@ -6,3 +6,8 @@
          "infixes" : ["450", "500"]
      }
  }
+
+training_input-image-_450.nc
+training_output-image-_450.nc
+training_input-image-_500.nc
+training_output-image-_500.nc