diff --git a/language_interpolation/state_space_network.py b/language_interpolation/state_space_network.py
index 699439a..a26bbc3 100644
--- a/language_interpolation/state_space_network.py
+++ b/language_interpolation/state_space_network.py
@@ -1,5 +1,6 @@
 # Originally copied from
 # https://github.com/johnma2006/mamba-minimal
+# and then modified to work with high order layers
 
 """Simple, minimal implementation of Mamba in one file of PyTorch.