improve zsh completion