diff --git a/strings/word_occurrence.py b/strings/word_occurrence.py
index 4acfa41adf11..f594fb0b3170 100644
--- a/strings/word_occurrence.py
+++ b/strings/word_occurrence.py
@@ -1,26 +1,20 @@
 # Created by sarathkaul on 17/11/19
 # Modified by Arkadip Bhattacharya(@darkmatter18) on 20/04/2020
-from collections import defaultdict
-
 
 def word_occurence(sentence: str) -> dict:
     """
-    >>> from collections import Counter
-    >>> SENTENCE = "a b A b c b d b d e f e g e h e i e j e 0"
-    >>> occurence_dict = word_occurence(SENTENCE)
-    >>> all(occurence_dict[word] == count for word, count
-    ...     in Counter(SENTENCE.split()).items())
-    True
-    >>> dict(word_occurence("Two  spaces"))
+    >>> word_occurence("Two  spaces")
     {'Two': 1, 'spaces': 1}
     """
-    occurrence: dict = defaultdict(int)
+    from collections import defaultdict
+    occurrence = defaultdict(int)
     # Creating a dictionary containing count of each word
     for word in sentence.split():
         occurrence[word] += 1
-    return occurrence
+    return dict(occurrence)
 
 
 if __name__ == "__main__":
-    for word, count in word_occurence("INPUT STRING").items():
-        print(f"{word}: {count}")
+    from doctest import testmod
+
+    testmod()