zshrc: change cursor colour depending on vi mode