Updated zsh prompt to highlight current directory, code to be more readable