mom-58 Include wind stree correction terms
mom-59 Use correct axes in diagnostics