diff --git a/assignment1/uz_framework/image.py b/assignment1/uz_framework/image.py
index a0db909..ea4d842 100644
--- a/assignment1/uz_framework/image.py
+++ b/assignment1/uz_framework/image.py
@@ -213,6 +213,9 @@ def get_image_bins(image: Union[npt.NDArray[np.float64],  npt.NDArray[np.uint8]]
     return counts / np.sum(counts)
 
 def apply_mask_on_image(image: Union[npt.NDArray[np.float64],  npt.NDArray[np.uint8]], mask: npt.NDArray[np.uint8]):
+    """
+    Accepts image and applys mask to image
+    """
     image = image.copy()
 
     mask = np.expand_dims(mask, axis=2)