When color variation is selected, only show images of that color variation

Hi there,

Im stuck at the moment and im trying to achieve this:
I have this great plugin working together with the “variation-swatches-for-woocommerce” plugin.

Everything works fine, but what im trying to achieve is that when a user clicks on a color variation, only the product images that contain that color variation are shown.

Now it only adds the class “current-thumbnail” to one image. But i would like to “filter” out all the images that contain the current selected color variation. And hide the other thumbnails.

Can you please help me witht his? Thanks in advance!

The page I need help with: [log in to see the link]