Add conditional for zsh prompt