Improve zsh git prompt