Adding K stock and first stage regression