Update restrict to branch instructions