diff --git a/docs/api.md b/docs/api.md
index 8254ff4..e04cf58 100644
--- a/docs/api.md
+++ b/docs/api.md
@@ -3,6 +3,7 @@
 
 class App:
     def __init__(self, options=None):
+    def register(self, extension):
     def template(self, template_engine):
     def json_serializer(self, json_serializer):
     def static(self, route, directory):
diff --git a/src/tests.py b/src/tests.py
index eb41c7a..1a0ceb1 100644
--- a/src/tests.py
+++ b/src/tests.py
@@ -41,13 +41,11 @@ app.listen(
 )
 app.run()
 
-# uws_websocket_upgrade_handler
-# uws_generic_method_handler
+
 # uws_websocket_drain_handler
 # uws_websocket_subscription_handler
 # uws_websocket_open_handler
 # uws_websocket_message_handler
 # uws_websocket_pong_handler
 # uws_websocket_ping_handler
-# uws_websocket_close_handler
-# uws_websocket_subscription_handler
\ No newline at end of file
+# uws_websocket_close_handler
\ No newline at end of file