Add oh-my-zsh support