working on fixing gradient of velocity field