diff --git a/deepof/models.py b/deepof/models.py
index 7bf5eb540311485b2d8fb106a27422ad843e5543..86f8888f4ea6cd56de75a074a6a48b598c8211c4 100644
--- a/deepof/models.py
+++ b/deepof/models.py
@@ -800,3 +800,4 @@ class SEQ_2_SEQ_GMVAE:
 #       - Check usefulness of stateful sequential layers! (stateful=True in the LSTMs)
 #       - Investigate full covariance matrix approximation for the latent space! (details on tfp course) :)
 #       - Explore expanding the event dims of the final reconstruction layer
+#       - Think about gradient penalty to avoid mode collapse (as in WGAN-GP)