diff --git a/count_words.py b/count_words.py
index acfa02f..af93828 100644
--- a/count_words.py
+++ b/count_words.py
@@ -1,4 +1,25 @@
-paragraph = paragraph.lower()
-paragraph = paragraph.translate(str.maketrans("", "", string.punctuation))
+import string
+from collections import Counter
 
-wordList = paragraph.split()
+def count_words(paragraph) -> int:
+  paragraph = paragraph.lower()
+  paragraph = paragraph.translate(str.maketrans("", "", string.punctuation))
+  wordList = paragraph.split()
+  counter = Counter(wordList)
+  return counter
+
+def main():
+  paragraph = """Nadia’s Garden Restaurant is the creation of husband and wife team Nadia and Timothy Arbore. 
+    Their American-infused, Italian-based, organically created, cuisine was inspired by Nadia’s papa, an immigrant from Italy, 
+    who shared his love of cooking with Nadia as a young girl. His focus on using fresh ingredients and family style dining were 
+    the inspiration for Nadia’s Garden Restaurant. Located in the heart of Main Streets Historic District, they are proud to be a 
+    part of a rich community. In 2011, the duo remodeled the restaurant and updated their menu to include newer and more diverse entrées
+     that could be made from local organic suppliers. Preservation of the building’s original layout has allowed them to create smaller, 
+     more intimate, dining spaces. Nadia and Timothy are committed to sharing their family history of cuisine, along with their new inspirations,
+      with their customers. Their passion for community, entertainment, and hospitality are found in every aspect of Nadia’s Garden Restaurant."""
+  print ("test test")
+  print(count_words(paragraph))
+
+if __name__ == "__main__":
+  main()
+  print ("test print")