main: Move TOKENS_CONTROL_FLOW handling into command word branch
main: Move TOKENS_CONTROL_FLOW handling into command word branch

No functional change.

P71/zsh-syntax-highlighting,zsh-users/zsh-syntax-highlighting,joshuarubin/dotfiles,phy1729/zsh-syntax-highlighting,joshuarubin/dotfiles,phy1729/zsh-syntax-highlighting,x-way/zsh-syntax-highlighting,P71/zsh-syntax-highlighting,joshuarubin/dotfiles,zsh-users/zsh-syntax-highlighting,joshuarubin/dotfiles,x-way/zsh-syntax-highlighting,stuha/wbr-zsh