This works elegantly! I just did a demo
If it counts at all I’d vote for having the features of the image alignment/.grid buttons theme component.
If there was a feature for images I would like to have it would be a way to add a caption, In our community I advocate for attributing images, which is not the best use of the image description.