Improve docs

Closes #9