Add virtualenv support for prompt