diff --git a/CMakeLists.txt b/CMakeLists.txt
index 2143028..073ccc3 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -1,5 +1,5 @@
 cmake_minimum_required(VERSION 3.25)
-project(COSC-4P80-Assignment-2 VERSION 0.0.4)
+project(COSC-4P80-Assignment-2 VERSION 0.0.5)
 
 option(ENABLE_ADDRSAN "Enable the address sanitizer" OFF)
 option(ENABLE_UBSAN "Enable the ub sanitizer" OFF)
diff --git a/include/assign2/layer.h b/include/assign2/layer.h
index 028a025..03ec2d4 100644
--- a/include/assign2/layer.h
+++ b/include/assign2/layer.h
@@ -110,6 +110,8 @@ namespace assign2
             {
                 std::vector<Scalar> dw;
                 
+                
+                // this is close! i think the changes should be applied in the neuron since the slides show the change of weight PER NEURON PER INPUT
                 // δ(h)
                 if (is_output)
                 {