Added vorticity, but there are still some issues with it