improve for zsh