diff --git a/src/twomartens/masterthesis/aae/train.py b/src/twomartens/masterthesis/aae/train.py
index fabcb0a..9db111b 100644
--- a/src/twomartens/masterthesis/aae/train.py
+++ b/src/twomartens/masterthesis/aae/train.py
@@ -30,6 +30,10 @@ Functions:
     prepare_training_data(...): prepares the mnist training data
     train(...): trains the AAE models
 
+Todos:
+    - fix early stopping
+    - fix losses reaching exactly zero
+
 """
 
 import functools
@@ -621,7 +625,7 @@ def _normalize(feature: tf.Tensor, label: tf.Tensor) -> Tuple[tf.Tensor, tf.Tens
 
 if __name__ == "__main__":
     tf.enable_eager_execution()
-    inlier_classes = [0]
+    inlier_classes = [3]
     iteration = 1
     train_dataset, _ = prepare_training_data(test_fold_id=0, inlier_classes=inlier_classes,
                                              total_classes=10)