Add vi-mode in zsh