Add an alternative way to get the correct register-ed output of alu