[WIP] Improve docs