diff --git a/src/jaxns/framework/model.py b/src/jaxns/framework/model.py
index 3347f1b..d9cd019 100644
--- a/src/jaxns/framework/model.py
+++ b/src/jaxns/framework/model.py
@@ -19,6 +19,7 @@
     'Model'
 ]
 
+# TODO: Rewrite Model using context framework to avoid recompiling functions.
 
 class Model(BaseAbstractModel):
     """