Improve doc

Closes #12