Starting to remove the state_reg to replace it with separate regs for key and iv.