Using $VIRTUAL_ENV to set virtualenv prompt

Previous prompt was looking for virtualenv and setting the prompt.

virtualenv may not be on PATH, for example, when you use pythonbrew.

Seeing if $VIRTUAL_ENV is set would be good check for this.