save initial z0 location as zp[:,0] else, it gets saved to output netCDF file as NaN