added velocity gradient update