Add zsh function