How can I display multiple variant images with the Flex theme?

Store Design
1

Hi, I am trying to figure out how I can have multiple variant images so that when selecting a swatch it will only show the images appropriate to that product. I am using the Flex theme.

2

I also have the same question. Did you ever find a solution?