Cleaned up the zsh completion script