Enable VI mode for zsh