Improve oh-my-zsh