diff --git a/.idea/workspace.xml b/.idea/workspace.xml
index baa276e..ac5dcbc 100644
--- a/.idea/workspace.xml
+++ b/.idea/workspace.xml
@@ -1,17 +1,7 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <project version="4">
   <component name="ChangeListManager">
-    <list default="true" id="5955480a-c876-43d5-afd7-8717f51f413e" name="Default Changelist" comment="Lightning integration basic ae, dataloaders and dataset">
-      <change afterPath="$PROJECT_DIR$/networks/adverserial_auto_encoder.py" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/.idea/ae_toolbox_torch.iml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/ae_toolbox_torch.iml" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/dataset.py" beforeDir="false" afterPath="$PROJECT_DIR$/dataset.py" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/networks/basic_ae.py" beforeDir="false" afterPath="$PROJECT_DIR$/networks/auto_encoder.py" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/networks/basic_vae.py" beforeDir="false" afterPath="$PROJECT_DIR$/networks/variational_auto_encoder.py" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/networks/modules.py" beforeDir="false" afterPath="$PROJECT_DIR$/networks/modules.py" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/run_basic_ae.py" beforeDir="false" afterPath="$PROJECT_DIR$/run_models.py" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/run_basic_vae.py" beforeDir="false" />
-    </list>
+    <list default="true" id="5955480a-c876-43d5-afd7-8717f51f413e" name="Default Changelist" comment="" />
     <option name="EXCLUDED_CONVERTED_TO_IGNORED" value="true" />
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -154,7 +144,7 @@
       <workItem from="1564587420277" duration="6891000" />
       <workItem from="1565364574595" duration="1092000" />
       <workItem from="1565592214301" duration="53660000" />
-      <workItem from="1565793671730" duration="30373000" />
+      <workItem from="1565793671730" duration="30478000" />
     </task>
     <task id="LOCAL-00001" summary="Lightning integration basic ae, dataloaders and dataset">
       <created>1565793753423</created>
@@ -170,7 +160,14 @@
       <option name="project" value="LOCAL" />
       <updated>1565958589041</updated>
     </task>
-    <option name="localTasksCounter" value="3" />
+    <task id="LOCAL-00003" summary="Done: AE, VAE, AAE&#10;ToDo: Double AAE, Visualization&#10;All Modularized">
+      <created>1565987964760</created>
+      <option name="number" value="00003" />
+      <option name="presentableId" value="LOCAL-00003" />
+      <option name="project" value="LOCAL" />
+      <updated>1565987964760</updated>
+    </task>
+    <option name="localTasksCounter" value="4" />
     <servers />
   </component>
   <component name="TypeScriptGeneratedFilesManager">
@@ -191,7 +188,8 @@
   </component>
   <component name="VcsManagerConfiguration">
     <MESSAGE value="Lightning integration basic ae, dataloaders and dataset" />
-    <option name="LAST_COMMIT_MESSAGE" value="Lightning integration basic ae, dataloaders and dataset" />
+    <MESSAGE value="Done: AE, VAE, AAE&#10;ToDo: Double AAE, Visualization&#10;All Modularized" />
+    <option name="LAST_COMMIT_MESSAGE" value="Done: AE, VAE, AAE&#10;ToDo: Double AAE, Visualization&#10;All Modularized" />
   </component>
   <component name="XDebuggerManager">
     <breakpoint-manager>