Add zsh/oh-my-zsh handling