diff --git a/src/easyvae/autoencoder.py b/src/easyvae/autoencoder.py
index f6f9566..d913699 100644
--- a/src/easyvae/autoencoder.py
+++ b/src/easyvae/autoencoder.py
@@ -96,7 +96,7 @@ class ClassicalAutoencoder(AAutoencoder):
                 f"Type: {__class__.__name__}",
                 "Encoder:",
                 f"{self.encoder}",
-                "Decoder:"
+                "Decoder:",
                 f"{self.decoder}"
             ))
 
@@ -133,6 +133,15 @@ class VariationalAutoencoder(AAutoencoder):
         super().__init__(*args, **kwargs)
         self.sampler = SampleLayer(self.encoder.out_size, self.lr, Identity())
 
+    def __str__(self):
+        return "\n".join((
+                f"Type: {__class__.__name__}",
+                "Encoder:",
+                f"{self.encoder}",
+                "Decoder:",
+                f"{self.decoder}"
+            ))
+
     def loss(self, data_set: list[np.ndarray]) -> float:
         loss = 0
         for x in data_set: