[AdgLDim] Using vector instead of matrix in update_shifts()