Add branch testing