Add VIRTUAL_ENV support to setup.py