diff --git a/src/aliby/utils/vis_tools.py b/src/aliby/utils/vis_tools.py
index c6d4b7de7a483056741a150121f2cdfd59a02b30..c8937d5dfba3dc88f6ec1a234718e19c6e30267e 100644
--- a/src/aliby/utils/vis_tools.py
+++ b/src/aliby/utils/vis_tools.py
@@ -2,8 +2,8 @@
 """
 Visualisation tools useful to generate figures cell pictures and figures from scripts.
 """
-
 import typing as t
+from copy import copy
 
 import numpy as np
 
@@ -140,3 +140,15 @@ def long_side_vertical(arr):
     if np.subtract(*arr.shape):
         result = arr.T
     return result
+
+
+def rescale(arr):
+    arr_min = arr.min()
+    arr_max = arr.max()
+    return (arr - arr_min) / (arr_max - arr_min)
+
+
+def crop_mask(img: np.ndarray, mask: np.ndarray):
+    img = copy(img).astype(float)
+    img[~mask] = np.nan
+    return img