Support prompt for both bash and zsh