1 files changed, 27 insertions, 11 deletions

diff --git a/keyboards/atreus/atreus.h b/keyboards/atreus/atreus.h
index 8e868ecc0..889a162a2 100644
--- a/keyboards/atreus/atreus.h
+++ b/keyboards/atreus/atreus.h
@@ -1,10 +1,28 @@
-#ifndef ATREUS_H
-#define ATREUS_H
+/* Copyright 2019
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+#pragma once
 
 #include "quantum.h"
-#include "matrix.h"
-#include "backlight.h"
-#include <stddef.h>
+#define ___ KC_NO
+
+#ifdef KEYBOARD_atreus_astar
+    #include "astar.h"
+#elif KEYBOARD_atreus_teensy2
+    #include "teensy2.h"
+#endif
 
 // This a shortcut to help you visually see your layout.
 // The first section contains all of the arguements
@@ -16,10 +34,8 @@
   k30, k31, k32, k33, k34, k35, k36, k37, k38, k39, k3a, k3b \
 ) \
 { \
-        { k00, k01, k02, k03, k04, KC_NO, k05, k06, k07, k08, k09 }, \
-        { k10, k11, k12, k13, k14, KC_NO, k15, k16, k17, k18, k19 }, \
-        { k20, k21, k22, k23, k24, k35,   k25, k26, k27, k28, k29 }, \
-        { k30, k31, k32, k33, k34, k36,   k37, k38, k39, k3a, k3b } \
+        { k00, k01, k02, k03, k04, ___, k05, k06, k07, k08, k09 }, \
+        { k10, k11, k12, k13, k14, ___, k15, k16, k17, k18, k19 }, \
+        { k20, k21, k22, k23, k24, k35, k25, k26, k27, k28, k29 }, \
+        { k30, k31, k32, k33, k34, k36, k37, k38, k39, k3a, k3b } \
 }
-
-#endif