diff --git a/src/twomartens/masterthesis/aae/__init__.py b/src/twomartens/masterthesis/aae/__init__.py
index 68f09f6..7e5b620 100644
--- a/src/twomartens/masterthesis/aae/__init__.py
+++ b/src/twomartens/masterthesis/aae/__init__.py
@@ -14,4 +14,16 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 
-"""masterthesis.aae: contains code for adversarial auto encoders"""
+"""
+Provides an AAE implementation.
+
+Modules:
+    ``model``: provides the keras models of the AAE implementation
+    
+    ``train``: provides functionality to train the AAE
+    
+    ``util``: provides helper functionality for visualization
+
+Todos:
+    - make the implementation compatible with the YCB Video dataset
+"""