diff --git a/documentation/MistakeDefinitions.txt b/docs/MistakeDefinitions.txt
similarity index 100%
rename from documentation/MistakeDefinitions.txt
rename to docs/MistakeDefinitions.txt
diff --git a/umbra/filereader.py b/umbra/filereader.py
index 9ce3f469cd49665052e0509fd137581323e3cea3..220ec569d4b2950094be49b0421ebedcd68eb96e 100644
--- a/umbra/filereader.py
+++ b/umbra/filereader.py
@@ -124,7 +124,7 @@ class TxtWriter(FileWriter):
         source_results, shadow_results, info = results
         final = np.empty(0)
         for i in range(len(source_results)):
-            source_word = source_results.words[i]
+            source_word = source_results[i]
             res = (str(source_word), str(source_word.shadowed)+"\n")
             final = np.append(final, res)
         final = np.append(final, "Total: "+str(info[0]))
diff --git a/umbra/mistake_finder.py b/umbra/mistake_finder.py
index 62285e9b474c4e2bba6fef63546661d052f65679..8ee6013efc2d599cf5e93d90ba0c9d8deb2fd026 100644
--- a/umbra/mistake_finder.py
+++ b/umbra/mistake_finder.py
@@ -72,14 +72,15 @@ class MistakeFinder:
 
 
     # TODO: WordNet integration. Function below should be usable already!
-    # def _check_semantic_mistake(self, shd_word):
-    #     shd_string = shd_word.word
-    #     src_string = shd_word.source.word
-    #     if model.seman_check.semantically_related(src_string, shd_string):
-    #         shd_word.mistake = Mistake.SEMANTIC
-    #         src_word.mistake = Mistake.SEMANTIC
-    #         return True
-    #     return False
+    def _check_semantic_mistake(self, shd_word):
+        shd_string = shd_word.word
+        src_string = shd_word.source.word
+        if model.seman_check.semantically_related(src_string, shd_string):
+            shd_word.mistake = Mistake.SEMANTIC
+            src_word.mistake = Mistake.SEMANTIC
+            return True
+        return False
+
+    def _check_phonetic_mistake(self, shd_word):
+        pass
 
-    # def _check_phonetic_mistake(self, shd_word):
-    #