ZSH: use widget that shows when terminal is in VI mode