Added uv_comm to calculate_uv, fixed pressure indexing