Rewrite if statements for zsh prompt