prelim module support

kmk/extensions/__init__.py

@@ -8,12 +8,12 @@ class Extension:
     def enable(self, keyboard):
         self._enabled = True
 
-        self.on_runtime_enable(self, keyboard)
+        self.on_runtime_enable(keyboard)
 
     def disable(self, keyboard):
         self._enabled = False
 
-        self.on_runtime_disable(self, keyboard)
+        self.on_runtime_disable(keyboard)
 
     # The below methods should be implemented by subclasses
 
@@ -32,7 +32,7 @@ class Extension:
         '''
         raise NotImplementedError
 
-    def after_matrix_scan(self, keyboard, matrix_update):
+    def after_matrix_scan(self, keyboard):
         '''
         Return value will be replace matrix update if supplied
         '''