Improve docs

Closes #15