[DOC] Improve documentation