diff --git a/src/lib/key-functions/public/basic.c b/src/lib/key-functions/public/basic.c
index fad8ea6..8f5927f 100644
--- a/src/lib/key-functions/public/basic.c
+++ b/src/lib/key-functions/public/basic.c
@@ -52,6 +52,7 @@ void kbfun_toggle(void) {
  * active
  */
 void kbfun_transparent(void) {
+  // TODO maybe re-implement this cleaner?
   main_arg_trans_key_pressed = true;
   main_arg_layer_offset++;
   main_arg_layer = main_layers_peek(main_arg_layer_offset);
diff --git a/src/main.c b/src/main.c
index 5b8e201..83feb7b 100644
--- a/src/main.c
+++ b/src/main.c
@@ -108,15 +108,7 @@ int main(void) {
 
 // ----------------------------------------------------------------------------
 
-/* ----------------------------------------------------------------------------
- * Layer Functions
- * ----------------------------------------------------------------------------
- * We keep track of which layer is foremost by placing it on a stack.  Layers
- * may appear in the stack more than once.  The base layer will always be
- * layer-0.
- *
- * Implemented as a fixed size stack.
- * ------------------------------------------------------------------------- */
+// layer functions
 
 // new array version
 
@@ -188,6 +180,16 @@ void main_layers_disable(uint8_t layer) {
   }
 }
 
+// return layer offset elements below the top
+uint8_t main_layers_peek(uint8_t offset) {
+  // TODO remove this
+  if (offset <= layers_head) {
+    return layers_stack[layers_head - offset].layer;
+  }
+
+  return 0; // default, or error
+}
+
 // execute the keypress or keyrelease function (if it exists) of the key at the current possition
 void main_exec_key(void) {
   void (*key_function)(void) =
@@ -216,24 +218,6 @@ void main_exec_key(void) {
 
 // TODO remove all this
 
-/*
- * peek()
- *
- * Arguments
- * - 'offset': the offset (down the stack) from the head element
- *
- * Returns
- * - success: the layer-number of the requested element (which may be 0)
- * - failure: 0 (default) (out of bounds)
- */
-uint8_t main_layers_peek(uint8_t offset) {
-  if (offset <= layers_head) {
-    return layers_stack[layers_head - offset].layer;
-  }
-
-  return 0; // default, or error
-}
-
 /*
  * push()
  *