Add in CAT branch selection