diff --git a/assignment3/uz_framework/image.py b/assignment3/uz_framework/image.py
index 5bafaff..baf5317 100644
--- a/assignment3/uz_framework/image.py
+++ b/assignment3/uz_framework/image.py
@@ -598,9 +598,8 @@ def find_edges_nms(image: Union[npt.NDArray[np.float64], npt.NDArray[np.uint8]],
 
 
     derivative_magnitude, derivative_angles = gradient_magnitude(image, sigma)
-
-    #reduced_magnitude = np.zeros_like(derivative_magnitude)
     reduced_magnitude = derivative_magnitude.copy()
+    # Set low elements to 0
     reduced_magnitude[(derivative_magnitude <= theta)] = 0