Save function the right way

This approach works for both bash and zsh. It should be possible now to use
env.sh + virtualenv in any sequence.

Change-Id: If0ba5932b857e86684038d77f832b36dd8555d99