Improve docs

Closes #48