Improve docs

Add docs with the same structure as ``richenum``.