move branch logic into EX stage