Improve oh-my-zsh integration