Added some branch logic to the register_file