diff --git a/u4/word_count.py b/u4/word_count.py
index 387f8ad..c81e967 100644
--- a/u4/word_count.py
+++ b/u4/word_count.py
@@ -1,8 +1,9 @@
 def count_words(phrase):
-    slovnik1 = []
+    slovnik1 = {}
     slova = phrase.lower().split()  
     for gns in slova:
-        cisteslovo = gns.strip(".,!:;?'\"")   
+        cisteslovo = gns.strip(".,!:;?'\"")  
         if cisteslovo:
-            slovnik1.append(cisteslovo) 
+            slovnik1[cisteslovo] = slovnik1.get(cisteslovo, 0) + 1
+
     return slovnik1