diff --git a/led_control.h b/led_control.h
index 2e182222..5f42edfa 100644
--- a/led_control.h
+++ b/led_control.h
@@ -18,10 +18,10 @@
 
 static const int key_led_map[4][16] = 
 {
-    {3,4,11,12,19,20,27,28,     35,39,40,47,48,55,56,63},
-    {2,5,10,13,18,21,26,29,     34,38,41,46,49,54,57,62},
-    {1,6,9,14, 17,22,25,30,     33,37,42,45,50,53,58,61},
-    {0,7,8,15,16,23,24,31,      32,36,43,44,51,52,59,60},
+    {3,4,11,12,19,20,26,27,     36,37,43,44,51,52,59,60},
+    {2,5,10,13,18,21,31,28,     35,32,42,45,50,53,58,61},
+    {1,6,9,14, 17,22,25,29,     34,38,41,46,49,54,57,62},
+    {0,7,8,15,16,23,24,30,      33,39,40,47,48,55,56,63},
 };
 
 
@@ -34,7 +34,7 @@ void set_led_mode(int mode);
 void next_led_mode();
 
 byte key_to_led(byte row, byte col);
-void set_key_color(byte row, byte col);
+void set_key_color(byte row, byte col, cRGB color);
 cRGB get_key_color(byte row, byte col);
 
 void led_compute_breath();