Moved prompt setting to a separate function for zshrc