port _prompt_dir function to zsh