BUG: Make crop_img work on 4D img