Improve docs
Merge pull request #21 from pikesley/refactor

Improve docs