Add pip completion to zsh