Add argument completion for zsh