Two small issues regarding this great implementation: 1) Should add here `int` around the index (in case the index comes out as a `float`. https://github.com/cynricfu/multi-band-blending/blob/7dae1133919ffd43528ab30b45b17de3218a76d1/multi_band_blending.py#L39 2) Should add warning/assert regarding shapes of images that are supposed to be the same