Modified the EX stage a little bit. not done yet.