I would combine `simple_3d` with `simple_3d_mask` into one function (supply a keyword argument `with_mask=True`)