Constrain to diagnoals with shift.