Minor tweaks to branch selector