diff --git a/adv_control/control_ctrlora.py b/adv_control/control_ctrlora.py
index 03e192f..5b3fbf4 100644
--- a/adv_control/control_ctrlora.py
+++ b/adv_control/control_ctrlora.py
@@ -1,3 +1,5 @@
+# Core code adapted from CtrLoRA github repo:
+# https://github.com/xyfJASON/ctrlora
 import torch
 from torch import Tensor