Improve the zsh task