Add vi mode for zsh