Adding the branch testing code.