Set the vi mode indicator in zsh