Add ability to edit zsh prompt in vim