Re-add vi mode back to zsh