Mod .zshrc for python