Better virtualenv support in the prompt