Improve support for rbenv

Some of this was taken from:

  https://github.com/sectioneight/dotfiles/blob/master/.zshrc#L32