diff --git a/bin/main.py b/bin/main.py
index 2aff7bcde..eb044c07d 100644
--- a/bin/main.py
+++ b/bin/main.py
@@ -1,5 +1,6 @@
 # coding=utf-8
 # Copyright 2020 The Google Research Authors.
+# Modifications copyright 2024 Henry Addison
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/src/ml_downscaling_emulator/score_sde_pytorch_hja22/losses.py b/src/ml_downscaling_emulator/score_sde_pytorch_hja22/losses.py
index d02b6c158..19b76bd95 100644
--- a/src/ml_downscaling_emulator/score_sde_pytorch_hja22/losses.py
+++ b/src/ml_downscaling_emulator/score_sde_pytorch_hja22/losses.py
@@ -1,5 +1,6 @@
 # coding=utf-8
 # Copyright 2020 The Google Research Authors.
+# Modifications copyright 2024 Henry Addison
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -12,6 +13,9 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+#
+# Modifications to the original work have been made by Henry Addison
+# to allow for sampling
 
 """All functions related to loss computation and optimization.
 """
diff --git a/src/ml_downscaling_emulator/score_sde_pytorch_hja22/models/cncsnpp.py b/src/ml_downscaling_emulator/score_sde_pytorch_hja22/models/cncsnpp.py
index 7b6428193..b1179f60f 100644
--- a/src/ml_downscaling_emulator/score_sde_pytorch_hja22/models/cncsnpp.py
+++ b/src/ml_downscaling_emulator/score_sde_pytorch_hja22/models/cncsnpp.py
@@ -1,5 +1,6 @@
 # coding=utf-8
 # Copyright 2020 The Google Research Authors.
+# Modifications copyright 2024 Henry Addison
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -12,6 +13,9 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+#
+# Significant modifications to the original work have been made by Henry Addison
+# to allow for conditional modelling
 
 # pylint: skip-file
 
diff --git a/src/ml_downscaling_emulator/score_sde_pytorch_hja22/models/utils.py b/src/ml_downscaling_emulator/score_sde_pytorch_hja22/models/utils.py
index 204bde8db..b559a5ba2 100644
--- a/src/ml_downscaling_emulator/score_sde_pytorch_hja22/models/utils.py
+++ b/src/ml_downscaling_emulator/score_sde_pytorch_hja22/models/utils.py
@@ -1,5 +1,6 @@
 # coding=utf-8
 # Copyright 2020 The Google Research Authors.
+# Modifications copyright 2024 Henry Addison
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -12,6 +13,9 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+#
+# Modifications to the original work have been made by Henry Addison
+# to allow for conditional modelling.
 
 """All functions and modules related to model definition.
 """
diff --git a/src/ml_downscaling_emulator/score_sde_pytorch_hja22/run_lib.py b/src/ml_downscaling_emulator/score_sde_pytorch_hja22/run_lib.py
index b51672bd5..18f28ef4a 100644
--- a/src/ml_downscaling_emulator/score_sde_pytorch_hja22/run_lib.py
+++ b/src/ml_downscaling_emulator/score_sde_pytorch_hja22/run_lib.py
@@ -1,5 +1,6 @@
 # coding=utf-8
 # Copyright 2020 The Google Research Authors.
+# Modifications copyright 2024 Henry Addison
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -12,6 +13,11 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+#
+# Significant modifications to the original work have been made by Henry Addison
+# to allow for conditional modelling, location-specific parameters,
+# removal of tensorflow dependency, tracking for training via Weights and Biases
+# and MLFlow, and iterating by epoch using PyTorch DataLoaders
 
 # pylint: skip-file
 """Training and evaluation for score-based generative models. """
diff --git a/src/ml_downscaling_emulator/score_sde_pytorch_hja22/sampling.py b/src/ml_downscaling_emulator/score_sde_pytorch_hja22/sampling.py
index 99cc74fa2..28ee1c7a8 100644
--- a/src/ml_downscaling_emulator/score_sde_pytorch_hja22/sampling.py
+++ b/src/ml_downscaling_emulator/score_sde_pytorch_hja22/sampling.py
@@ -1,5 +1,6 @@
 # coding=utf-8
 # Copyright 2020 The Google Research Authors.
+# Modifications copyright 2024 Henry Addison
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -12,6 +13,9 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+#
+# Modifications to the original work have been made by Henry Addison
+# to allow for sampling
 
 # pylint: skip-file
 # pytype: skip-file