Enable Vi mode in ZSH