DOC: improve documentation