Added magic for py virtualenv shell prompt