zsh: Improve virtualenv prompt