added calculation of terms in vorticity budget. Still need to integrate and plot