Refactor

We need to run this code inside of
`Bundler.with_clean_env` or the yard-tomdoc plugin
isn't loaded properly.
Merge pull request #3 from site5/refactor

Refactor