Got it! I like this one better because you can avoid the need to customize the css and paste the short code each time… but whatever works for you is great!
You can manage the columns in the dashboard area under woocommere > woo variations table
Make sure you check the box next to images! My product just pulled the product image, but you can assign a different one for each variation. Let me know if that works!