Thank You, the code is working!
Also, can we have the color option as variant image in variant picker?