Enable vi-mode in zsh with custom bindings