* 🔨 Extract prepare mask logic

* 🐛 return Image instead of nparray

* 🐛 rename variable so that same variable always have same type

* 🎨 nits