Improve docs, fixes #3