diff --git a/autostack/main.py b/autostack/main.py
index ded64e7..9dd6185 100644
--- a/autostack/main.py
+++ b/autostack/main.py
@@ -113,7 +113,3 @@ def main():
                         custom_query()
                         break
                     continue
-                    
-if __name__ == "__main__":
-    # execute only if run as a script
-    main()
\ No newline at end of file
diff --git a/autostack/option_features/features.py b/autostack/option_features/features.py
index 53ce278..45a6d27 100644
--- a/autostack/option_features/features.py
+++ b/autostack/option_features/features.py
@@ -13,7 +13,12 @@
 
 def custom_query():
     '''
-    Allows the user to submit a custom query to search on Stack Overflow.
+    Searches Stack Overflow using a custom search query from the user.
+
+    Called after three "no"s from the user. This function prompts the user
+    for a string to serach on Stack Overflow. Using the same for loop as
+    main, questions and accepted answers are printed to the terminal for the
+    user to view.
     '''
 
     print('Enter a custom query: ', end='')