Coding A Rainbow Vortex With Python