Scripts to calculate moment diagnostics for the stratospheric polar vortex