diff --git a/yoink.py b/yoink.py
index fea248e..df1bf40 100644
--- a/yoink.py
+++ b/yoink.py
@@ -16,9 +16,9 @@ class Yoink:
         Constructor.
 
         Args:
-            file: string, Shodan keys
-            filt: string, Filter for Shodan keys
-            output: string, Output file to write
+            file: str, Shodan keys
+            filt: str, Filter for Shodan keys
+            output: str, Output file to write
         """
 
         self.file = file
@@ -34,7 +34,7 @@ class Yoink:
         Add the key to a tree.
 
         Args:
-            key: string, Current valid key
+            key: str, Current valid key
             query_credits: int, Query credits left
             scan_credits: int, Scan credits left
             plan: str, Current plan