modified for better vi-mode zsh usage