Use vi mode for zsh