1 files changed, 19 insertions, 13 deletions
diff --git a/keyboards/jj40/jj40.c b/keyboards/jj40/jj40.c
index d5add2a6b..6044e83fd 100644
--- a/keyboards/jj40/jj40.c
+++ b/keyboards/jj40/jj40.c
@@ -22,19 +22,25 @@ along with this program.  If not, see <http://www.gnu.org/licenses/>.
 #include "action_layer.h"
 #include "quantum.h"
 
-__attribute__ ((weak))
-void matrix_scan_user(void) {
-    /* Nothing to do here... yet */
+#include "i2c.h"
+
+// custom RGB driver
+extern rgblight_config_t rgblight_config;
+void rgblight_set(void) {
+  if (!rgblight_config.enable) {
+    for (uint8_t i=0; i<RGBLED_NUM; i++) {
+      led[i].r = 0;
+      led[i].g = 0;
+      led[i].b = 0;
+    }
+  }
+
+  i2c_init();
+  i2c_send(0xb0, (uint8_t*)led, 3 * RGBLED_NUM);
 }
 
-void matrix_init_kb(void) {
-
-  // Call the keymap level matrix init.
-  matrix_init_user();
-
-  // Set our LED pins as output
-  DDRB |= (1<<6);
+__attribute__ ((weak))
+void matrix_scan_user(void) {
+    rgblight_task();
+    /* Nothing else for now. */
 }
-
-void matrix_init_user(void) {
-}
-\ No newline at end of file