Vi mode for zsh