Update zsh prompt code