Improve zsh config