More clever zsh completion