diff --git a/src/twomartens/masterthesis/aae/train.py b/src/twomartens/masterthesis/aae/train.py
index 13c7ed4..284eee7 100644
--- a/src/twomartens/masterthesis/aae/train.py
+++ b/src/twomartens/masterthesis/aae/train.py
@@ -399,6 +399,9 @@ def extract_batch(data: np.ndarray, it: int, batch_size: int) -> tfe.Variable:
 
 if __name__ == "__main__":
     tf.enable_eager_execution()
-    train_summary_writer = summary_ops_v2.create_file_writer('./summaries/train')
+    inlier_classes = [0]
+    iteration = 1
+    train_summary_writer = summary_ops_v2.create_file_writer(
+        './summaries/train/number-' + str(inlier_classes[0]) + '/' + str(iteration))
     with train_summary_writer.as_default(), summary_ops_v2.always_record_summaries():
-        train_mnist(folding_id=0, inlier_classes=[0], total_classes=10)
+        train_mnist(folding_id=0, inlier_classes=inlier_classes, total_classes=10)